argsort函数的作用,举例说明
时间: 2024-03-25 07:36:02 浏览: 68
浅述python中argsort()函数的实例用法
5星 · 资源好评率100%
argsort函数是numpy库中的一种排序方法,主要作用是返回数组值从小到大的索引值。
举个例子,假设我们有一个一维数组a=[3, 1, 4, 2, 5],我们可以使用argsort函数对其进行排序,代码如下:
```python
import numpy as np
a = np.array([3, 1, 4, 2, 5])
idx = np.argsort(a)
print(idx)
```
输出结果为:
```python
[1 3 0 2 4]
```
这里的idx数组表示排好序后各个元素在原数组中的索引值,即a[1]最小,a[4]最大,a[1]在原数组中的索引值为1,所以idx[0]=1,以此类推。因此,我们可以通过idx数组来获取排好序后的数组。例如,我们可以通过a[idx]来获取排好序后的数组:
```python
sorted_a = a[idx]
print(sorted_a)
```
输出结果为:
```python
[1 2 3 4 5]
```
这样,我们就得到了从小到大排好序的数组。
阅读全文