【发布时间】：2017-03-05 22:37:47
【问题描述】：
我有一个长度为 100 的 numpy 数组 A，还有一个长度为 4 的数组 B。
我想得到一个长度为 100 的掩码，当A 的元素等于B 的任何元素时，此掩码的值为 TRUE，否则为 FALSE。我怎样才能有效地做到这一点？
【问题讨论】：
标签： python numpy slice indices
【发布时间】：2017-03-05 22:37:47
【问题描述】：
试试：
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])
我想这就是你想要的
【讨论】：
-
谢谢，虽然我宁愿不使用 for 循环。显然这有效`np.in1d'