熊猫查找基于价值

Question

我有以下数据框：

Date best a b c d 
1990 a  5 4 7 2 
1991 c  10 1 2 0 
1992 d  2 1 4 12 
1993 a  5 8 11 6 

我想提出一个数据帧如下：

Date best value 
1990 a  5 
1991 c  2 
1992 d  12 
1993 a  5 

所以我期待基于另一行寻找价值值使用列名称。例如，第二个df中的1990的值应该从第一个df查找“a”，第二个行应该从第一个df中查找“c”（= 2）。

任何想法？

Answer 1

您创建一个查询功能，在您的数据帧逐行调用apply，这不是大DFS虽然

In [245]: 

def lookup(x): 
    return x[x.best] 
df['value'] = df.apply(lambda row: lookup(row), axis=1) 
df 
Out[245]: 
    Date best a b c d value 
0 1990 a 5 4 7 2  5 
1 1991 c 10 1 2 0  2 
2 1992 d 2 1 4 12  12 
3 1993 a 5 8 11 6  5 

Answer 2

有一个内置的lookup功能，可以处理这种情况非常有效（按行/列查找）。我不知道它是如何优化的，但可能比适用解决方案更快。

In [9]: df['value'] = df.lookup(df.index, df['best']) 

In [10]: df 
Out[10]: 
    Date best a b c d value 
0 1990 a 5 4 7 2  5 
1 1991 c 10 1 2 0  2 
2 1992 d 2 1 4 12  12 
3 1993 a 5 8 11 6  5