在每一行中列出数据框为空/空的列名称

Question

我有一个带有空/空值的数据框。
通过执行以下操作，我可以轻松获取空值每一行的计数：

df['NULL_COUNT'] = len(df[fields] - df.count(axis=1)

这将放置NULL字段中的列数NULL_COUNT。

如果为空，是否可以用相同的方式将列标题写入另一个字段？

df['NULL_FIELD_NAMES'] = "<some query expression>"

例：

df = pd.DataFrame([range(3), [0, np.NaN, 0], [0, 0, np.NaN], range(3), range(3)], columns=['A', 'B', 'C'])

在上面的df中，第二行应具有df['NULL_FIELD_NAME'] = 'B'，第三行应具有df['NULL_FIELD_NAME'] = 'C'

Answer 1

您可以使用：

df['new'] = (df.isnull() * df.columns.to_series()).apply(','.join,axis=1).str.strip(',')

另一个解决方案：

df['new'] = df.apply(lambda x: ','.join(x[x.isnull()].index),axis=1)

样品：

df = pd.DataFrame([range(3), [np.NaN, np.NaN, 0], [0, 0, np.NaN], range(3), range(3)], 
                  columns=['A', 'B', 'C'])
print (df)
     A    B    C
0  0.0  1.0  2.0
1  NaN  NaN  0.0
2  0.0  0.0  NaN
3  0.0  1.0  2.0
4  0.0  1.0  2.0

df['new'] = df.apply(lambda x: ','.join(x[x.isnull()].index),axis=1)
print (df)
     A    B    C  new
0  0.0  1.0  2.0     
1  NaN  NaN  0.0  A,B
2  0.0  0.0  NaN    C
3  0.0  1.0  2.0     
4  0.0  1.0  2.0