pandas：打印列名称，缺少值

Question

我正在尝试打印或获取缺少值的列名称列表。例如

data1 data2 data3  
1     3     3  
2     NaN   5  
3     4     NaN

我想获取[‘data2’，’data3’]。我写了以下代码：

print('\n'.join(map(
    lambda x : str(x[1])
    ,(filter(lambda z: z[0] != False, zip(train.isnull().any(axis=0), train.columns.values)))
)))

它运作良好，但我认为应该是更简单的方法。

Answer 1

df.isnull().any()生成一个布尔数组（如果该列具有缺失值，则为True，否则为False）。您可以使用它索引到df.columns：

df.columns[df.isnull().any()]

将返回缺少值的列的列表。

df = pd.DataFrame({'A': [1, 2, 3], 
                   'B': [1, 2, np.nan], 
                   'C': [4, 5, 6], 
                   'D': [np.nan, np.nan, np.nan]})

df
Out: 
   A    B  C   D
0  1  1.0  4 NaN
1  2  2.0  5 NaN
2  3  NaN  6 NaN

df.columns[df.isnull().any()]
Out: Index(['B', 'D'], dtype='object')

df.columns[df.isnull().any()].tolist()  # to get a list instead of an Index object
Out: ['B', 'D']