python - Numpy - 找到所有元素都是有限的行

Question

我有数据存储在一个有 n 行和 p 列的 numpy 数组中。

我想检查哪些行是完全有限的，并将此信息存储在 bool 数组中，以便在某处用作掩码。

我已经解决了 p=2 的情况，但我想解决所有情况

我的代码是这样的:

raw_test = np.array([[0, numpy.NaN], [0, 0], [numpy.NaN, numpy.NaN]])
test = np.isfinite(raw_test)
def multiply(x):
    return x[0] * x[1]

numpy.apply_along_axis(multiply, 1, test)

Answer 1

您可以使用 numpy.isnan 来检查哪些项目是 NaN，然后找到所有 True 的行的索引使用 numpy.all 和 numpy.where。

>>> np.isnan(raw_test)
array([[False,  True],
       [False, False],
       [ True,  True]], dtype=bool)
>>> np.all(np.isnan(raw_test), axis=1)
array([False, False,  True], dtype=bool)
>>> np.where(np.all(np.isnan(raw_test), axis=1))[0]
array([2])