如何获取一个数组中包含在另一个数组中的值的所有索引？

Question

假设我们有两个简单的 1d numpy 数组：

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

现在我想要的是获取所有可能的索引，其中数组 b 中的每个值都包含在数组 a 中。

我们可以做例如以下：

idx = np.where(np.any(a.reshape((-1, 1))==b, axis=1))[0]

idx 是 array([0, 1, 3, 4, 8])（这是我真正想要的）。

现在我真的很好奇在 numpy 或任何其他库中是否已经有类似的函数来解决这个问题（我相信已经存在一个）。否则我会坚持现在的工作方法。

Answer 1

与np.isin/np.in1d -

np.flatnonzero(np.isin(a,b))
# or np.flatnonzero(np.in1d(a,b))

对于排序的b，使用np.searchsorted 可能会更高效-

sidx = np.searchsorted(b,a)
sidx[sidx==len(b)] = len(b)-1
out = np.flatnonzero(b[sidx]==a)

如果b 未排序，则对其进行排序，然后使用它代替b。