Получение индексов элементов, находящихся в другом списке, в numpy ⇐ Python

[Anonymous]

У меня есть два массива 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). Существует ли эквивалент для массивов значений?

Подробнее здесь: https://stackoverflow.com/questions/278 ... t-in-numpy