假设我有一个 numpy 数组
a = np.array([0, 8, 25, 78, 68, 98, 1])
和一个掩码数组b = [0, 1, 1, 0, 1]
有没有一种简单的方法来获取以下数组:
[8, 25, 68] - 这是原始数组中的第一个、第二个和第四个元素。这对我来说听起来像是一个面具。
我尝试过的最明显的方法是a[b]，但这并没有产生理想的结果。 之后我尝试查看 masked operations in numpy但看起来它引导我走向错误的方向。
最佳答案
如果a和b都是numpy数组，并且b严格是1和0:
>>> a[b.astype(np.bool)]
array([ 8, 25, 68])
应该指出的是，这仅在极小的情况下才会明显更快，并且范围比 @falsetru 的答案要有限得多:
a = np.random.randint(0,2,5)
%timeit a[a==1]
100000 loops, best of 3: 4.39 µs per loop
%timeit a[a.astype(np.bool)]
100000 loops, best of 3: 2.44 µs per loop
对于较大的情况:
a = np.random.randint(0,2,5E6)
%timeit a[a==1]
10 loops, best of 3: 59.6 ms per loop
%timeit a[a.astype(np.bool)]
10 loops, best of 3: 56 ms per loop
关于python - numpy 从数组中获取掩码，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/18765904/