在Numpy数组中查找多个值

Question

我正在寻找一个numpy函数来查找某个值在矢量（xs）内找到的索引。这些值在另一个数组（ys）中给出。返回的索引必须遵循ys的顺序。

在代码中，我想用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] 

有没有快速的方法？

感谢

Answer 1

对于大阵列xs和ys，你将需要改变的基本方法这成为快。如果你是排序xs罚款，然后一个容易的选择是使用numpy.searchsorted()：

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

如果要保持xs原来的顺序是非常重要的，你可以用这个办法了，但是你需要记住的原始指数：

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