Buscar valores múltiples dentro de una matriz Numpy

Question

Estoy buscando una función numpy para encontrar los índices en los que se encuentran ciertos valores dentro de un vector (xs). Los valores se dan en otra matriz (ys). Los índices devueltos deben seguir el orden de ys.

En el código, quiero reemplazar la lista de comprensión a continuación por una función numpy.

>> import numpy as np 
>> xs = np.asarray([45, 67, 32, 52, 94, 64, 21]) 
>> ys = np.asarray([67, 94]) 
>> ndx = np.asarray([np.nonzero(xs == y)[0][0] for y in ys]) # <---- This line 
>> print(ndx) 
[1 4] 

¿Hay una manera rápida?

Gracias

Answer 1

para grandes matrices xs y ys, que tendrían que cambiar el enfoque básico para que esto sea rápido. Si usted está muy bien con la clasificación xs, a continuación, una opción fácil es utilizar numpy.searchsorted():

xs.sort() 
ndx = numpy.searchsorted(xs, ys) 

Si es importante para mantener el orden original de xs, puede utilizar este enfoque, también, pero hay que recordar la índices originales:

orig_indices = xs.argsort() 
ndx = orig_indices[numpy.searchsorted(xs[orig_indices], ys)]