Numpy中的argsort()函数

np.argsort

argsort函数返回的是数组值从小到大的索引值

一维数组排序：

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

>>> np.argsort(-x)  #从大到小排序
array([0, 2, 1], dtype=int64)

二维数组：

x = np.array([[0,3,8],[7,6,5],[7,1,4]])
y = np.argsort(x,axis=0)#按列排序  相对应会返回从小到大的索引（相对于所在行）
print(y)

输出：
[[0 2 2]
 [1 0 1]
 [2 1 0]]
 
y = np.argsort(x,axis=1)#按行排序  相对应会返回从小到大的索引
输出：
[[0 1 2]
 [2 1 0]
 [1 2 0]]