Python 输入ID时打印行的用户定义函数

Python 输入ID时打印行的用户定义函数,python,pandas,portfolio,Python,Pandas,Portfolio,嗨,伙计们,我不知道该怎么开始。我在这里附上了一张照片。我可以问一下如何定义一个用户函数,在这个函数中我输入id,我将得到整行吗 谢谢你们的帮助。问题是: 用于一行数据帧与双[]一起使用,仅用于系列一个[]: df= pd.read_csv('portfolios.csv') df1 = df.set_index('id');df df1 你好,杰兹,你能举个例子吗?非常感谢你! df = pd.DataFrame({'id':[1,2,3,4,5,6],

嗨,伙计们,我不知道该怎么开始。我在这里附上了一张照片。我可以问一下如何定义一个用户函数,在这个函数中我输入id,我将得到整行吗

谢谢你们的帮助。问题是:


用于一行
数据帧
与双
[]
一起使用,仅用于
系列
一个
[]

df= pd.read_csv('portfolios.csv') 
df1 = df.set_index('id');df
df1



你好,杰兹,你能举个例子吗?非常感谢你!
df = pd.DataFrame({'id':[1,2,3,4,5,6],
                   'B':[4,5,4,5,5,4],
                   'C':[7,8,9,4,2,3],
                   'D':[1,3,5,7,1,0],
                   'E':[5,3,6,9,2,4],
                   'F':list('aaabbb')})

df1 = df.set_index('id')
print (df1)
    B  C  D  E  F
id               
1   4  7  1  5  a
2   5  8  3  3  a
3   4  9  5  6  a
4   5  4  7  9  b
5   5  2  1  2  b
6   4  3  0  4  b
def get_line1(x, val):
    row = x.loc[[val]]
    return row

def get_line2(x, val):
    row = x.loc[val]
    return row
out = get_line1(df1, 5)
print (out)
    B  C  D  E  F
id               
5   5  2  1  2  b

out = get_line2(df1, 5)
print (out)
B    5
C    2
D    1
E    2
F    b
Name: 5, dtype: object