python - 倒花式索引

Question

有了一个数组和这个数组的掩码，使用花哨的索引，很容易只选择掩码对应的数组的数据。

import numpy as np

a = np.arange(20).reshape(4, 5)
mask = [0, 2]
data = a[:, mask]

但是有没有一种快速的方法可以把数组中所有不属于掩码的数据都选出来(即掩码就是我们要拒绝的数据)？我试图找到一个通过中间 bool 数组的通用解决方案，但我确信确实有更简单的方法。

mask2 = np.ones(a.shape)==1
mask2[:, mask]=False
data = a[mask2].reshape(a.shape[0], a.shape[1]-size(mask))

谢谢

Answer 1

看看numpy.invert、numpy.bitwise_not、numpy.logical_not，或者更简洁的~mask。 (在这种情况下，他们都做同样的事情。)

举个简单的例子:

import numpy as np

x = np.arange(10)
mask = x > 5

print x[mask]
print x[~mask]