如果满足条件，则用列名更新行值

Question

我试图对熊猫数据框执行以下操作：

输入数据：

Char   A    B
Nan    0    1
Nan    1    0

输出：

Char   
B
A

如您所见，如果所述列的值为 1，我想用列名填充 'Char' 中的值。我应该如何解决这个问题？

Answer 1

您可以使用

df['Char'] = df[['A','B']].idxmax(axis=1)
df

出去：

  Char  A  B
0    B  0  1
1    A  1  0