Python—numpy.argsort()

函数将数组的值从小到大排序后，并按照其相对应的索引值输出.

一维数组：

>>> a = array([3,1,2])
>>> argsort(a)
array([1, 2, 0]) 

二维数组：

>>> b = array([[1,2],[2,3]])
>>> argsort(b,axis=1) #按行排序
array([[0, 1],
       [0, 1]])
>>> argsort(b,axis=0) #按列排序
array([[0, 0],
       [1, 1]])