python - 如何从两个数组中找到匹配的元素和索引？

Question

例如，

a = [1, 1, 2, 4, 4, 4, 5, 6, 7, 100]
b = [1, 2, 2, 2, 2, 4, 5, 7, 8, 100]

我可以使用以下方法找到匹配的元素:

np.intersect1d(a,b)

输出:

array([  1,   2,   4,   5,   7, 100])

那么，如何分别获取数组 a 和 b 中匹配元素的索引？
IDL 中有一个函数 "match" - https://www.l3harrisgeospatial.com/docs/match.html
Python中有类似的功能吗？

Answer 1

在 return_indices 中使用 numpy.intersect1d :

intersect, ind_a, ind_b = np.intersect1d(a,b, return_indices=True)

输出:

intersect
# array([  1,   2,   4,   5,   7, 100])
ind_a
# array([0, 2, 3, 6, 8, 9], dtype=int64)
ind_b
# array([0, 1, 5, 6, 7, 9], dtype=int64)

然后可以重复使用，如:

np.array(a)[ind_a]
np.array(b)[ind_b]

array([  1,   2,   4,   5,   7, 100])