python - 检索包含 NaN 值的行的索引

Question

我尝试为包含 NaN 值的每一行检索所有相应的索引。

      A    B    C    D
0  11.4  1.3  2.0  NaN
1  11.4  1.3  NaN  NaN
2  11.4  1.3  2.8  0.7
3  NaN   NaN  2.8  0.7

期望的结果:[0,1,3]

我正在寻找一种 pandas 方法来处理我庞大的数据集。提前致谢

Answer 1

用于根据 boolean indexing 的 index 值进行更好的性能过滤:

idx = df.index[df.isnull().any(axis=1)].tolist()
print (idx)
[0, 1, 3]

解释:

首先比较NaN的值:

print (df.isnull())
       A      B      C      D
0  False  False  False   True
1  False  False   True   True
2  False  False  False  False
3   True   True  False  False

然后通过 DataFrame.any 检查每行是否至少有一个:

print (df.isnull().any(axis=1))
0     True
1     True
2    False
3     True
dtype: bool