问numpy中内置的二进制搜索算法，这类似于np.searchsorted吗？
EN

Stack Overflow用户

提问于 2022-11-12 21:55:12

回答 1查看 23关注 0票数 0

我有一个主numpy数组a，还有另一个numpy数组b。我要做的是遍历b的每个元素，并检查该元素是否存在于a中。请记住，a和b都是相当庞大的，所以我想避免O(N)搜索时间。

我知道np.searchsorted(a,b)存在，但这提供了一个索引，我需要在其中放置b。这并不能告诉我b的一个元素是否马上就出现在a中。

我的问题是，如果来自b的元素存在于a中，那么numpy中是否有一个二进制搜索算法，它只报告真假？我知道我可以写一个，但如果有一个矢量化的，随时可用，我可以节省一些时间。

如有任何建议，将不胜感激！

回答已采纳

发布于 2022-11-12 22:13:19

完成排序搜索之后，可以检查这些索引处的元素是否等于b中的元素：

a = numpy.array([1,2,3,4,7])
b = numpy.array([1,4,5,7])
x = numpy.searchsorted(a,b)
boolean_array = a[x] == b

searchsorted indicates that使用默认的side = 'left'确保：a[i-1] < v <= a[i]，因此如果a[i]等于b中的对应元素，那么它将给出所需的匹配。

票数 2

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/74416885

复制

相似问题

问numpy中内置的二进制搜索算法，这类似于np.searchsorted吗？EN