python - 数据框查找行以返回索引

Question

我有一个数据框

Alpha Bravo Charlie Final
10     20    30       30
15     25    35       25

我想创建一个带有标题的新列，这样它将是:

Alpha Bravo Charlie Final NewColumn
10     20    30       30  Charlie
15     25    35       25  Bravo

我试过了，但没有给出正确的结果:

df['NewColumn'] = df.lookup(df.index, df.Final)

编辑:

我尝试了这两个答案并且它有效，但是因为我有许多其他列，有没有办法可以将查找功能限制为仅 Alpha Bravo 和 Charlie 列？ (它正在返回我其他列)

Answer 1

这是一个使用广播比较的 100% numpy 向量化解决方案:

i = df.values
j = df.Final.values

df['NewColumn'] = df.columns[(i == j[:, None]).argmax(1)]

df

   Alpha  Bravo  Charlie  Final NewColumn
0     10     20       30     30   Charlie
1     15     25       35     25     Bravo