python - 检查 : for each element in array check if it exists in another array 的高效实现

Question

我有两个 400K 元素范围内的大型一维 numpy 数组。我需要检查数组 A 中的每个元素是否存在于数组 B 中。我使用了 in1d但它似乎太慢并且需要很多时间。我想知道是否有任何方法可以加快速度？

A = np.array([1,2,3,4,5,6,7]) 
B = np.array([3,4,7])
result = np.in1d(A, B, invert=True)
result
>> array([ True,  True, False, False,  True,  True, False]

Answer 1

尝试将 B 转换为更适合搜索的结构(哈希集或排序集)