python - 在 pandas df 中返回列名称的最有效方法

Question

我有一个 pandas df，其中包含 4 个不同的列。对于每行行都有一个重要的值。我想返回显示值的列名称。因此，对于下面的 df，我想在标记值 2 时返回 Column 名称。

d = ({
    'A' : [2,0,0,2],     
    'B' : [0,0,2,0],
    'C' : [0,2,0,0],            
    'D' : [0,0,0,0], 
    })

df = pd.DataFrame(data=d)

输出:

   A  B  C  D
0  2  0  0  0
1  0  0  2  0
2  0  2  0  0
3  2  0  0  0

所以它会是A，C，B，A

我通过以下方式执行此操作

m = (df == 2).idxmax(axis=1)[0]

然后更改行。但这效率不高。

我还希望将 pandas df 的输出作为 Series 生成

Answer 1

使用DataFrame.dot:

df.astype(bool).dot(df.columns).str.cat(sep=',')

或者，

','.join(df.astype(bool).dot(df.columns))

'A,C,B,A'

或者，作为列表:

df.astype(bool).dot(df.columns).tolist()
['A', 'C', 'B', 'A']

...或系列:

df.astype(bool).dot(df.columns)

0    A
1    C
2    B
3    A
dtype: object