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

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

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

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

6
задан sega_sai 25 April 2012 в 21:14
поделиться