Numpy学习教程——argsort用法及参数详解_numpy argsort参数

numpy.argsort(a, axis=-1, kind=None, order=None)：用于给数组排序，返回值为从小到大元素index的值

假设一个数组a为[0,1,2,20,67,3],使用numpy.argsort(a),返回值应为[0,1,2,5,3,4]
a:要排序的数组
axis:排序方向（axis=0:按列排，axis=1:按行排, axis=None:将数组扁平化）

axis=None:先把数组扁平化变成一维数组，再进行排序，假设一个数组z为[[0,1,19],[20,67,3]],当使用numpy.argsort(z,axis=None)时，系统先用flatten()把z扁平化为[0,1,19,20,67,3],再进行排序，得到的结果应为为[0,1,5,2,3,4]

同时axis有时也可以为负数，通常用于多维数组：
假设一个数组K形状为[19,19,5,80]：
K.argmax(a,axis= 0 or -4) returns (19,5,80) with values from 0 to 18
K.argmax(a,axis= 1