我正在尝试运行一个函数来获取与数据框中的2对应的列名
输入数据框：
df1
A|B|C|D
1|0|2|4
2|9|1|3
输出数据框：
A|B|C|D|new|
1|0|2|4|C|
2|9|1|3|A|
我已经制作了函数，但是由于某种原因，我遇到了TypeError ：（“无法散列的类型：'列表'”，“发生在索引0”）
def colname(x):
b = (df1.loc[x]==2)
return b.idxmax()
df1["new"]=df1.apply(colname,axis=1)