numpy-np.searchsorted的用法

最新推荐文章于 2024-11-29 23:45:41 发布

zlibo丶

最新推荐文章于 2024-11-29 23:45:41 发布

阅读量742

点赞数

CC 4.0 BY-SA版权

分类专栏： numpy学习文章标签： python numpy

本文链接：https://blog.youkuaiyun.com/Libo_Learner/article/details/113106819

numpy学习专栏收录该内容

6 篇文章

订阅专栏

本文详细介绍了NumPy库中的searchsorted函数使用方法。该函数可以在已排序的数组中找到特定值的插入位置，并且提供了参数调整选项以适应不同场景的需求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

官方文档：https://numpy.org/doc/stable/reference/generated/numpy.searchsorted.html

官方给出的说明是：

即：查询给定数据在原始数组中的插入位置，而不改变原始数组的排列顺序。

代码测试：

print(np.searchsorted([1,2,3,4,5], 3))
print(np.searchsorted([1,2,3,4,5], 3, side='right'))
print(np.searchsorted([1,2,3,4,5], [-10, 10, 2, 3]))

"""
输出：
2
3
[0 5 1 2]
"""

给定已经排序完成的数组，在数组中插入'3'和新数组'[-10, 10, 2, 3]'，而不改变原始数组的排列顺序。返回插入数值的下标。

当给定的数组乱序排列时，可借助参数'sorter'完成对给定数组的排列之后，再寻找插入给定数值的下标位置，如下：

print(np.searchsorted([1,2,6,3,4,5], 4))  # 不调用sorter参数
print(np.searchsorted([1,2,6,3,4,5], 4, sorter=np.argsort([1,2,6,3,4,5])))  # 调用sorter参数

"""
输出：
4
3
"""

参考：

https://blog.youkuaiyun.com/qq_33757398/article/details/89876088