np.argsort 参数
时间: 2024-06-18 17:03:04 浏览: 129
浅析python中numpy包中的argsort函数的使用
5星 · 资源好评率100%
np.argsort()是NumPy中的一个函数,它返回排序后的索引。具体来说,它可以接收一个数组作为输入,然后返回一个按照元素大小排序后的原数组的索引值数组。默认情况下,它会按照升序排列,也可以通过传递可选参数来指定降序排列。
np.argsort()的参数包括:
- axis:默认值为-1,表示按照最后一维度进行排序。
- kind:可选值包括"quicksort"、"mergesort"和"heapsort"。默认值为"quicksort",表示使用快速排序算法进行排序。
- order:用于指定排序时要比较的字段。仅在输入数组是结构化数组时才可用。
以下是一个例子:
```python
import numpy as np
arr = np.array([2, 4, 3, 1, 5])
sorted_indices = np.argsort(arr)
print(sorted_indices) # [3 0 2 1 4]
```
上述代码中,输入的数组是[2, 4, 3, 1, 5],返回的结果是[3, 0, 2, 1, 4]。也就是说,原数组中最小值是1,它在排序后的数组中的索引是3;最大值是5,在排序后的数组中的索引是4。
阅读全文