python numpy中的 argsort_python中numpy多维数组argsort

将元素从小到大排列，然后提取索引；
axis = 0 按行排序
axis = 1 按列排序

例如：
原始数据
3,0,2,1,4,5
排序之后
0,1,2,3,4,5
排序之后的在原始数据中的索引，如：0在原始数据中索引是1
1,3,2,0,4,5

>>> a = array([2, 3, 1])
>>> argsort(a)    # 升序排列
array([2, 0, 1])  
>>> argsort(-a)   # 降序排列 
array([1, 0, 2])
1
2
3
4
5

>>> b = array([[2, 1, 3], [1, 2, 3]])
>>> b
array([[2, 1, 3],
       [1, 2, 3]])
>>> argsort(b, axis = 0) # 按列排序
array([[1, 0, 0],
       [0, 1, 1]])
>>> argsort(b, axis = 1) # 按行排序
array([[1, 0, 2],
       [0, 1, 2]])

>>> argsort(b)   # 默认行排序
array([[1, 0, 2],
       [0, 1, 2]])
>>> argsort(-b)
array([[2, 0, 1],
       [2, 1, 0]])
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17