python - 只输出 Pandas 行的某些部分

Question

我有这个数据框

       X    Y   Z    A
0      1    2   0    4   
1      0    2   3    0 
2      1    0   3    4     
3      1    0   0    4  
4      1    0   0    0   

我想将它保存为这种格式的文本文件(输出所有行，但排除任何包含 0 的列)

1,2,4
2,3
1,3,4
1,4
1

这可以用 for 循环输出，但对于大数据帧来说效率很低。有没有更快的方法来做到这一点？

Answer 1

替换 + str.cat

df.astype(str).replace('0',np.nan).apply(lambda x :x.str.cat(sep=','),1)
Out[525]: 
0    1,2,4
1      2,3
2    1,3,4
3      1,4
4        1
dtype: object