numpyargsort用法

numpy.argsort函数是numpy库中的一个排序函数，用于返回数组中元素的索引，以便按升序或降序排序。

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

参数说明： a：要排序的数组。 axis：指定排序的轴，axis=0表示按列排序，axis=1表示按行排序。 kind：排序算法，可选参数，包括'quicksort'、'mergesort'、'heapsort'等，默认为'quicksort'。 order：排序的字段，可选参数。

函数返回值： 返回排序后的索引。

示例代码： import numpy as np

a = np.array([3, 1, 4, 2]) print("原数组：", a) print("升序排序后的索引：", np.argsort(a)) print("降序排序后的索引：", np.argsort(-a))

输出结果： 原数组： [3 1 4 2] 升序排序后的索引： [1 3 0 2] 降序排序后的索引： [2 0 3 1]