Python Numpy 非零

Question

所以我得到了这个形状为 (31641600,2) 的 numpy 数组，其中有一些零值，如果不是很多的话。

我们称这个数组为 X。

正在做:

print len(X)
>>> 31641600

然后做:

X = X[np.nonzero(X)]
print len(X)
>>> 31919809

不明白为什么第二个更大。关于Documentation它说应用上述方法应该只返回非零值，因此 X 的长度应该更小。

有什么想法吗？谢谢。

Answer 1

这可能是因为 len(X) 只返回 X 沿第一个轴的长度。当你做的时候

X = X[np.nonzero(X)]

你得到一个一维数组，所以如果你在 X 中的零少于 50%，len(X) 将会增加。

考虑:

In [1]: import numpy as np

In [2]: X = np.zeros((42, 2))

In [3]: X[:, 0] = 1

In [4]: X[0, 1] = 1

In [5]: len(X)
Out[5]: 42

In [6]: len(X[np.nonzero(X)])
Out[6]: 43

那是因为 X[np.nonzero(X)] 是一个包含 43 个一的数组:

In [7]: X[np.nonzero(X)].shape
Out[7]: (43,)

根据评论更新:如果实际上您想要第一个元素非零的所有对，您可以这样做:

X = X[ X[:, 0] != 0 ]