numpy.argsort的用法介绍

参考文章：https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.argsort.html

numpy.argsort(aaxis=-1kind='quicksort'order=None)

返回数组排序后对应的下标。kind是排序算法，axis是排序的轴。

Parameters:	a Array to sort. axis Axis along which to sort. The default is -1 (the last axis). If None, the flattened array is used. kind Sorting algorithm. order a
Returns:	index_array aaa[index_array]a.

举例：

1、一维数组

>>> x = np.array([3, 1, 2]) >>> np.argsort(x) array([1, 2, 0])

2、二维数组

>>> x = np.array([[0, 3], [2, 2]]) >>> x array([[0, 3],        [2, 2]])

>>> np.argsort(x, axis=0) array([[0, 1],        [1, 0]])

>>> np.argsort(x, axis=1) array([[0, 1],        [0, 1]])

文章来源: numpy.argsort的用法介绍