python - 如何构建高级切片的逻辑表达式

Question

我正在尝试找出一种更简洁的方法来执行以下操作:

import numpy

a = np.array([1,2,4,5,1,4,2,1])

cut = a == (1 or 2)
print cut

[ True False False False  True False False  True]

以上当然是一个简化的例子。表达式 (1 or 2) 可以很大也可以很复杂。首先，我想概括一下:

cutexp = (1 or 2)
cut = a == cutexp

也许，cutex 可以变成一个函数或其他东西，但我不确定从哪里开始寻找。

Answer 1

你也可以试试 numpy.in1d .说

>>> a = np.array([1,2,4,5,1,4,2,1])
>>> b = np.array([1,2]) # Your test array
>>> np.in1d(a,b)
array([ True,  True, False, False,  True, False,  True,  True], dtype=bool)