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)