Numpy Array: эффективно найти подходящие индексы

У меня есть два списка, один из которых огромный (миллионы элементов), другой несколько тысяч. Я хочу сделать следующее

<code>bigArray=[0,1,0,2,3,2,,.....]

smallArray=[0,1,2,3,4]

for i in len(smallArray):
  pts=np.where(bigArray==smallArray[i])
  #Do stuff with pts...
</code>

Выше работает, но медленно. Есть ли способ сделать это более эффективно, не прибегая к написанию чего-либо на C?

Ответы на вопрос(3)

Ваш ответ на вопрос