E-Ink 新闻日报

← 返回列表

NumPy 2.5 将 np.searchsorted 速度提升最高 25 倍

NumPy 2.5 通过向量化二分查找算法优化了 np.searchsorted,结合分支消除、批处理和缓存友好布局,在基准测试中实现高达 25 倍的加速,相较于此前纯 Python 或对数级查询的旧实现。该优化惠及 SciPy、scikit‑learn 等核心科学计算库,并与 Python Array API 标准中的并行实现相一致。

背景

np.searchsorted 是用于直方图分箱和区间查找的基础二分搜索工具,广泛服务于 Python 科学计算生态。通过向量化原语对其进行优化可为反复调用该操作的下游库带来显著性能提升。

来源
Lobsters
发布时间
2026年10月11日 07:25
评分
7.0 / 10