逐列获取非零和非nan元素

Question

import numpy as np

我有一个 numpy 数组如下：

data = np.array([[2,3,4,5,0,7,8,9,9,3,np.nan],
                 [3,3,6,5,0,7,9,9,9,3,1],
                 [2,6,4,5,0,7,9,9,9,4,1]])
print (data)

我想从所有数字（按列）不为 0 的地方获取数据；并且任何元素都不是 np.nan。

indices = np.any(data, axis=1)

print (data[indices])

预计是：

[[2 3 4 5 7 8 9 9 3]
 [3 3 6 5 7 9 9 9 3]
 [2 6 4 5 7 9 9 9 4]]

我尝试如下掩码。

mask = np.all(np.isnan(data) | np.equal(data, 0), axis=1)

print (data[~mask])



[[ 2.  3.  4.  5.  0.  7.  0.  9.  9.  3. nan]
 [ 3.  3.  6.  5.  0.  7.  9.  9.  9.  3.  1.]
 [ 2.  6.  4.  5.  0.  7.  9.  9.  9.  4.  1.]]

但不是预期的结果

Answer 1

沿列使用 np.any (axis=0)。将其与任何 np.isnan 结合使用 这将返回列索引，其中该列中的任何值都不为零且没有 nan。 然后使用这些索引原始数组。

mask = ~np.any(np.isnan(data),axis=0) &  np.any(data, axis=0)
indices=data[:,mask]

Answer 2

你可以这样做：

data[:,(~np.isnan(data)).all(axis=0)&~(data==0).any(axis=0)].astype("int")

输出：

[[2 3 4 5 7 8 9 9 3]
 [3 3 6 5 7 9 9 9 3]
 [2 6 4 5 7 9 9 9 4]]