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

我正在尝试对熊猫数据帧执行以下操作：

输入数据：

Char   A    B
Nan    0    1
Nan    1    0

输出：

Char   
B
A

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

您可以使用

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

输出：

你可以用

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

输出：

---

让我们看看你到目前为止都做了些什么。