python - 选择行值不为空的 pandas 数据框的列名

Question

这是一个示例数据框:-

 A      B    C
23     45  30
54     39  NaN
NaN    45  76
87     32  NaN

我想要一个列表(或列表列表)，其中包含行值不是 NaN 的列名。

预期输出:-

A B C
A B
B C
A B

这样做的正确方法是什么？感谢您的期待。

Answer 1

IIUC，你可以使用 .dot df.columns 与 df.notna() 的乘积:

df.notna().dot(df.columns+',').str.rstrip(',') #you can assign to a new column

---

0    A,B,C
1      A,B
2      B,C
3      A,B
dtype: object