基于值是否在某个其他列表中的Numpy掩码

Question

我已经搜索了高低,并且无法找到一种方法(可能我正在寻找错误的术语.)

我想根据每个值是否在其他列表中创建一个掩码(例如[True False False True True]).

a=np.array([11,12,13,14,15,16,17])
mask= a in [14,16,8] #(this doesnt work at all!)
#I would like to see [False False False True False True False]

到目前为止,我能想出的最好的是列表理解

mask = [True if x in other_list else False for x in my_numpy_array]

请告诉我,如果你知道一些秘密的调味品,用numpy和fast(计算)来做这个,因为这个列表在现实中是巨大的......

Answer 1

用途numpy.in1d():

In [6]: np.in1d(a, [14, 16, 18])
Out[6]: array([False, False, False,  True, False,  True, False], dtype=bool)