np.argsort()用法

numpy.argsort(a, axis=-1, kind=’quicksort’, order=None)
功能: 将矩阵a按照axis排序，并返回排序后的下标
参数: a:输入矩阵， axis:需要排序的维度
返回值: 输出排序后的下标

示例代码：

点击查看代码

a = np.array([[4, 2, 1], [2, 4, 3],[8, 3, 1]])
b = np.argsort(a, axis=0)
c = np.argsort(a, axis=1)
print(b,'\n\n', c)

out：

点击查看代码

[[1 0 0]
 [0 2 2]
 [2 1 1]] 

 [[2 1 0]
 [0 2 1]
 [2 1 0]]

当 axis=0 时，相当于“按每列，将所有行的数据排序”；当 axis=1 时，相当于“按每行，将所有列的数据排序”；这个axis的用法，与np.sum()的相同