更改 pandas 数据框

Question

我有一组数据如下：

name A B C
foo  1 0 0
bar  0 1 0 
coo  0 0 1

我正在尝试更改为如下表所示：

   name 
A  foo
B  bar
C  coo

我做过研究，但没有运气。我不确定在 python 中是否可行。我需要手动更改此表才能获得我想要的结果吗？

Answer 1

代码

out = df.set_index('name').idxmax()

出

A    foo
B    bar
C    coo
dtype: object