python - 如何删除 numpy.ndarray 中包含非数字值的所有行

Question

我以 numpy.ndarray 的形式读取数据集，并且某些值丢失(要么不存在，要么是 NaN，要么是字符串写为“NA”)。

我想清除包含此类条目的所有行。如何使用 numpy ndarray 做到这一点？

Answer 1

>>> a = np.array([[1,2,3], [4,5,np.nan], [7,8,9]])
array([[  1.,   2.,   3.],
       [  4.,   5.,  nan],
       [  7.,   8.,   9.]])

>>> a[~np.isnan(a).any(axis=1)]
array([[ 1.,  2.,  3.],
       [ 7.,  8.,  9.]])

并将其重新分配给 a .

说明:np.isnan(a)返回一个类似的数组 True哪里NaN , False别处。 .any(axis=1)减少 m*n数组到n具有逻辑性or对整行进行操作，~反转True/False和a[ ]仅选择原始数组中具有 True 的行括号内。