numpy.argsort解析

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

argsort(x,axis)

x=x:          按升序排序

x=-x:         按降序排序

axis=1:      按行排序

axis=0:      按列排序

一维的数组：

import numpy as np
x = np.array([3, 1, 2])
print(np.argsort(x))

#输出：[1 2 0]

二维的数组：

import numpy as np

x = np.array([[0, 3], [4, 2]])
print("----------输出x-----------")
print(x)
print("---------按列排序---------")
print(np.argsort(x, axis=0))    #按列排序
print("---------按行排序---------")
print(np.argsort(x, axis=1))    #按行排序

#输出：
----------输出x-----------
[[0 3]
 [4 2]]
---------按列排序---------
[[0 1]
 [1 0]]
---------按行排序---------
[[0 1]
 [1 0]]