numpy argsort
时间: 2023-11-10 07:18:46 浏览: 147
3 Numpy计算1
numpy.argsort 函数返回数组排序后的索引值。这个函数可以用于对数组进行排序,同时获取排序后元素的原始索引位置。
它的语法如下:
```python
numpy.argsort(a, axis=-1, kind='quicksort', order=None)
```
参数解释:
- a:要排序的数组。
- axis:默认为 -1,表示按最后一个轴排序。
- kind:排序算法的种类,默认为 'quicksort'。
- order:如果数组是结构化数组,则可以指定要排序的字段。
返回值:
返回一个按照指定轴排序后的数组索引。
举个例子,我们可以看一下如何使用 argsort 函数:
```python
import numpy as np
a = np.array([3, 1, 2])
indices = np.argsort(a)
print(indices)
```
输出结果为 [1 2 0],表示按照索引位置排序后的结果。在这个例子中,原始数组为 [3, 1, 2],经过排序后,得到的数组为 [1, 2, 3],对应的索引位置为 [1, 2, 0]。
阅读全文