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