python - for 循环以提取 pandas 中数据帧的 header

Question

我是python新手。我有一个如下所示的数据框:

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

如何编写 for 循环来收集每一行的列名。我希望我的结果集看起来像这样:

    A   B   C   D   E   Result
0   1   0   1   0   1   ACE
1   0   1   0   0   1   BE
2   0   1   1   1   0   BCD
3   1   0   0   1   0   AD
4   1   0   0   1   1   ADE

有人可以帮我吗？谢谢!

Answer 1

dot 函数就是为此目的而完成的，因为您需要矩阵与列名向量之间的矩阵点积:

df.dot(df.columns)
Out[5]: 
0    ACE
1     BE
2    BCD
3     AD
4    ADE

如果您的数据帧是数字，则首先通过针对 0 测试您的 df 来获取 bool 矩阵:

(df!=0).dot(df.columns)

PS: 只需将结果分配给新列即可

df['Result'] = df.dot(df.columns)

df
Out[7]: 
   A  B  C  D  E Result
0  1  0  1  0  1    ACE
1  0  1  0  0  1     BE
2  0  1  1  1  0    BCD
3  1  0  0  1  0     AD
4  1  0  0  1  1    ADE