在特定条件下显示Pandas数据帧中每一行的列名

Question

我正在使用Python Pandas Dataframe并尝试打印数据集中每行的列列表，假设每列可以有0或1的值。例如：

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

现在，我需要输出：

id   output
 0   A,B,C,D
 1   B,D
 2   A,B
 3   A

请注意，我需要准备一个通用功能，而不管列名或编号。

Answer 1

你可以这样做：

df = df.assign(output=df.dot(df.columns))

df[['output']]

id   output
 0   A,B,C,D
 1   B,D
 2   A,B
 3   A