获取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]),)