numpy的argsort函数用法
时间: 2023-09-09 13:10:20 浏览: 164
浅析python中numpy包中的argsort函数的使用
5星 · 资源好评率100%
numpy的argsort函数用于返回数组中元素排序后的索引值,可以按照升序或降序进行排序。
使用方法如下:
```python
numpy.argsort(a, axis=-1, kind='quicksort', order=None)
```
参数说明:
- `a`:待排序的数组
- `axis`:指定排序的轴,默认为最后一维
- `kind`:排序算法,可选值为`quicksort`、`mergesort`、`heapsort`,默认为`quicksort`
- `order`:排序时参考的字段,用于结构化数组
函数返回值是一个数组,数组的每个元素是原数组中元素的索引值。
示例代码:
```python
import numpy as np
a = np.array([3, 1, 4, 2])
idx = np.argsort(a)
print(idx)
```
输出结果为:
```
[1 3 0 2]
```
说明:
- 元素1是原数组中最小的,索引值为1
- 元素3是原数组中第二小的,索引值为3
- 元素0是原数组中第三小的,索引值为0
- 元素2是原数组中最大的,索引值为2
因此,对于原数组`a`,按照升序进行排序后的结果为`[a[1], a[3], a[0], a[2]]`,即`[1, 2, 3, 4]`。
阅读全文