numpy-np.searchsorted的用法

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

原创最新推荐文章于 2024-11-29 23:45:41 发布 · 854 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #numpy

numpy学习专栏收录该内容

6 篇文章

订阅专栏

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

官方文档：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