获取 numpy 中另一个列表中元素的索引

Question

我有两个 numpy 数组，我想获取第二个数组中第一个数组中所有元素的索引。例如：

import numpy as np
x = np.array([0,1,1,2,3,4,5,5])
y = np.array([1,3])
# want to get np.array([1,2,4]) 

如果

y

是一个标量，我就可以做

np.where(x == y)

。值数组是否有等效项？

Answer 1

您可以

numpy.where

与

numpy.in1d

:

>>> np.where(np.in1d(x, y))
(array([1, 2, 4]),)

Answer 2

新代码中推荐的方法是使用 isin 而不是

in1d

,

np.where(np.isin(x, y))

还给出 (array([1, 2, 4]),)