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

たぶん、cutexpは関数か何かに変えることができますが、どこから始めればよいのかわかりません。

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)

Answer 2

>>> (a == 2) | (a == 1)
array([ True,  True, False, False,  True, False,  True,  True], dtype=bool)