相关疑难解决方法(0)

是否有一个快速numpy函数用于返回较大数组中的索引列表,其中它匹配较小数组的值？较小的数组是~30M值,更大的是800M,所以我想避免一个for循环的numpy.where调用.

searchsorted的问题是它会返回结果,即使它们不是完全匹配,它只是给出最接近的索引,但我只想要有完全匹配的索引

而不是这个:

>>> a = array([1,2,3,4,5])
>>> b = array([2,4,7])
>>> searchsorted(a,b)
array([1, 3, 5])

我想要这个:

>>> a = array([1,2,3,4,5])
>>> b = array([2,4,7])
>>> SOMEFUNCTION(a,b)
array([1, 3])

编辑:较小和较大数组中的值集始终是唯一的并且已排序.