在python中显示基于0、1的列名

Question

我正在尝试在python中进行以下转换

来自

   A  B  C  D  E  
   x  1  0  0  0 
   y  0  1  1  0 
   z  1  0  0  1 

到

   x  B 
   y  C,D
   z  A,E

你有什么想法吗？

Answer 1

使用dot

进行检查

df=df.set_index('A')
df.dot(df.columns+',').str[:-1]
A
x      B
y    C,D
z    B,E
dtype: object