通过仅保留所有 arrays 中存在的数字来过滤一组 arrays？ [复制]

Question

这个问题在这里已经有了答案：

NumPy：比较两个 Arrays 中的元素（6 个答案）

去年关闭。

我有两个 arrays 充满了对象的 ID，我只想保留两个 arrays 中存在的对象的 ID。 例子：

a = np.array([1, 2, 3, 2, 4, 1])
b = np.array([3, 4, 5, 6])

因此，理想的 output 将是一个数组，该数组已通过 arrays a 和 b 进行搜索，并为我提供了一个仅包含两个列表中存在的对象的数组。 类似于：

c = np.array([3, 4])

该数组显示 ID 3 和 ID 4 都出现在两个列表中。 有没有办法做到这一点？

Answer 1

不要使用 arrays，因为您需要设置：

set(a) & set(b)

如果您需要数组中的结果：

np.array(list(set(a) & set(b)))