Python 获取numpy中其他列表中元素的索引

我有两个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）

。值数组是否有等效项？

您可以使用：

您可以使用：

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