按索引数据框搜索

时间:2018-05-30 09:20:27

标签: python dataframe

我有一个列Search,其中包含标题字符串:

Alpha Bravo Charlie Search SearchReturn
1      2      3     Alpha      1  
2      5      6     Charlie    6

我正在尝试为SearchReturn列使用excel Vlookup函数。

但是,我不知道如何创建SearchReturn列,你能建议吗?

2 个答案:

答案 0 :(得分:3)

对行标签和列标签使用pd.DataFrame.lookup

df['SearchReturn'] = df.lookup(df.index, df['Search'])

print(df)

   Alpha  Bravo  Charlie   Search  SearchReturn
0      1      2        3    Alpha             1
1      2      5        6  Charlie             6

答案 1 :(得分:1)

这样做:

df['SearchReturn'] = df.apply(lambda x: x[x['Search']], axis=1)