如何得到numpy数组的指标等于另一个数组的任何值？

Question

我有一个numpy数组A，长度为100，另一个数组为B，长度为4.

我想获得一个长度为100的掩码，当A的元素等于B的任何元素时，此掩码的值为TRUE，否则为FALSE。我怎样才能以有效的方式做到这一点？

Answer 1

尝试：

a = np.random.rand(100)
b = np.random.rand(4)
print([(element in b) for element in a])
print('\n')
print([(element in a) for element in b])

我认为这就是你想要的东西