np.searchsorted()
时间: 2023-10-13 13:07:06 浏览: 31
`np.searchsorted()` 是 NumPy 中用于在已排序数组中查找插入位置的函数。它的原型如下:
```python
np.searchsorted(sorted_array, values, side='left')
```
其中,`sorted_array` 表示已排序的一维数组,`values` 是要查找插入位置的一维数组,`side` 是可选参数,表示插入位置的方向,可以取 `'left'` 或者 `'right'`,默认为 `'left'`。
`np.searchsorted()` 返回的是一个一维数组,表示 `values` 中每个元素要插入到 `sorted_array` 中的位置。如果 `side` 参数为 `'left'`,则返回的是左侧插入位置,如果 `side` 参数为 `'right'`,则返回的是右侧插入位置。
举个例子:
```python
import numpy as np
arr = np.array([2, 3, 5, 7, 9])
indices = np.searchsorted(arr, [1, 5, 8])
print(indices) # 输出 [0 2 4]
```
在这个例子中,`arr` 是一个已排序的数组,`np.searchsorted(arr, [1, 5, 8])` 返回的是一个一维数组,表示元素 1、5、8 要插入到 `arr` 中的位置。在这个例子中,1 要插入到数组的最左侧,所以返回的是 0;5 要插入到数组的第 3 个位置,所以返回的是 2;8 要插入到数组的第 5 个位置,所以返回的是 4。