选择行值不为空的列名称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，您可以将df.columns的{​​{3}}乘以.dot：

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