Python numpy.argsort 排序索引

np.argsort可返回排序后的索引
一维列表：

>>> import numpy as np 
>>> a = [3,2,6,-1,5]
>>> np.argsort(a)
array([3, 1, 0, 4, 2], dtype=int64)

二维列表矩阵行排序

>>> b = [[2, -1, 5, 4],[3, 2, 0, 0],[-1, 6, -1, 2],[1, 1, 1, 1]]
>>> np.argsort(b)
array([[1, 0, 3, 2],
       [2, 3, 1, 0],
       [0, 2, 3, 1],
       [0, 1, 2, 3]], dtype=int64)

二维列排序

>>> np.argsort(b,axis=0)
array([[2, 0, 2, 1],
       [3, 3, 1, 3],
       [0, 1, 3, 2],
       [1, 2, 0, 0]], dtype=int64)