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)