python里的argsort函数
时间: 2023-06-08 14:09:14 浏览: 57
argsort函数是一个numpy库中的函数,它用于返回数组排序后的索引值,从小到大排列。它的函数定义为:numpy.argsort(a, axis=-1, kind='quicksort', order=None)。其中a是需要排序的数组,axis表示需要排序的维度,默认为最后一个维度,kind表示排序算法的类型,默认为快速排序算法,order表示按照指定列排序的列号。
相关问题
argsort函数python
argsort函数是NumPy的一个函数,用于对数组进行排序并返回排序后每个元素在原数组中对应的下标。
语法:numpy.argsort(a, axis=-1, kind=None, order=None)
参数说明:
- a:要排序的数组
- axis:排序的轴,默认为最后一维
- kind:排序算法,可选值为{'quicksort', 'mergesort', 'heapsort', 'stable'},默认为'quicksort'
- order:指定排序字段,用于结构化数组
返回值:返回一个数组,表示排序后每个元素在原数组中对应的下标。
例子:
```python
import numpy as np
a = np.array([3, 1, 4, 2])
idx = np.argsort(a)
print(idx) # [1 3 0 2]
```
上面的例子中,a数组被排序后为[1, 2, 3, 4],而idx数组则表示每个元素在原数组中的下标,即[1, 3, 0, 2]。
np.argsort函数用法
np.argsort函数是NumPy库中的一个函数,用于返回数组排序后的索引值。它可以按照指定的轴对数组进行排序,并返回排序后的索引值数组。
该函数的用法如下:
```python
numpy.argsort(a, axis=-1, kind=None, order=None)
```
参数说明:
- a:要排序的数组。
- axis:指定排序的轴,默认为-1,表示按照最后一个轴进行排序。
- kind:指定排序算法,默认为'quicksort',还可以选择'mergesort'和'heapsort'。
- order:如果数组是结构化数组,则可以指定按照哪个字段进行排序。
返回值:
- 返回一个数组,其中包含按照指定轴排序后的索引值。
示例代码:
```python
import numpy as np
arr = np.array([3, 1, 2])
sorted_indices = np.argsort(arr)
print(sorted_indices) # 输出:[1 2 0]
```
上述示例中,我们创建了一个一维数组arr,然后使用np.argsort函数对其进行排序,并将排序后的索引值存储在sorted_indices中。最后打印sorted_indices,输出为[1 2 0],表示原数组arr中的元素按照从小到大的顺序排列时,索引为1的元素最小,索引为2的元素次小,索引为0的元素最大。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)