Python NumPy searchsorted方法用法及代码示例

例子

搜索按 side=left 排序

要查找值 7(如果它已添加到数组 [6,7,8,9] 中)的索引：

np.searchsorted([6,7,8,9], 7)   # or with parameter side=left



1

这里，返回1，因为当值7插入到数组中时，它将被放置在第一个索引中(即其重复项的左侧)。

此外，重要的是我们的输入数组按升序排序，因为我们没有提供 sorter 参数。

搜索按 side=right 排序

要获取值的正确索引：

np.searchsorted([6,7,8,9], 7, side="right")



2

这里，返回2，因为当值7插入到数组中时，它将被放置在第二个索引中(即其重复项的右侧)。

传入值数组

我们还可以传递一个值数组，而不是只传递一个标量，如下所示：

np.searchsorted([2,3,4,5], [0,3,6])



array([0, 1, 4])

请注意如何逐一考虑每个元素 - 这就是值 6 的索引是 4 的原因。

传入分拣机

方法 searchsorted(~) 要求输入数组按升序排序。如果不是，我们需要传入一个一维数组，该数组按升序对输入数组进行排序，如下所示：

np.searchsorted([2,1,3,4], 2, sorter=[1,0,2,3])



1