python - 如何获取所有非空行值的列名？

Question

给定以下数据框

    A   B   C
0   1   Nan Nan
1  Nan  2   Nan
2  Nan  Nan 3

我想获取一个列，获取所有非空行的列名

预期结果

   A   B   C   D
0   1   2 Nan   [A,B]
1  Nan  2   4   [B,C]
2   4  Nan 3    [A,C]

有没有使用python的解决方案？

Answer 1

使用:

df['D'] = df.apply(lambda x: x.dropna().index.to_list(), axis=1)