Numpy Array: Encontre eficientemente os índices correspondentes

Eu tenho duas listas, uma das quais é enorme (milhões de elementos), os outros vários milhares. Eu quero fazer o seguinte

<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>

O acima funciona, mas é lento. Existe alguma maneira de fazer isso de forma mais eficiente, sem recorrer a escrever algo em C?

questionAnswers(3)

yourAnswerToTheQuestion