python - Numpy:删除所有 nan 或 0 值的行

Question

如果行 = nan，我想从表中删除所有值或 0 .

我知道有一种方法可以使用 pandas 来做到这一点，即 pandas.dropna(how = 'all')但我想要一个 numpy 方法来删​​除所有 nan 的行或 0 .

有没有有效的实现方式？

Answer 1

import numpy as np

a = np.array([
    [1, 0, 0],
    [0, np.nan, 0],
    [0, 0, 0],
    [np.nan, np.nan, np.nan],
    [2, 3, 4]
])

mask = np.all(np.isnan(a) | np.equal(a, 0), axis=1)
a[~mask]