Получение индексов элементов, находящихся в другом списке в 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]),)