argsort函数python

argsort函数是NumPy的一个函数，用于对数组进行排序并返回排序后每个元素在原数组中对应的下标。

语法：numpy.argsort(a, axis=-1, kind=None, order=None)

参数说明：
- a：要排序的数组
- axis：排序的轴，默认为最后一维
- kind：排序算法，可选值为{'quicksort', 'mergesort', 'heapsort', 'stable'}，默认为'quicksort'
- order：指定排序字段，用于结构化数组

返回值：返回一个数组，表示排序后每个元素在原数组中对应的下标。

例子：

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]。