Numpy学习之np.argsort()函数

对数组进行排序时，有时想返回排序后的索引，如返回数组中最小的三个值，这时可以用到np.argsort()函数，现在简单介绍一下这个函数。

np.argsort(a, axis=-1, kind=‘quicksort’, order=None)
Returns the indices that would sort an array.返回将对数组进行排序的索引。

代码示例：
一维情形：

x = np.random.randint(0,10,10)
print(x)
[9 2 2 7 1 5 1 5 1 2]

y = np.argsort(x)#默认从小到大排列，返回相应索引
print(y)
[4 6 8 1 2 9 5 7 3 0]
print(x[y])
[1 1 1 2 2 2 5 5 7 9]

y = np.argsort(-x)#对x，从大到小排列，返回相应索引
print(y)
[0 3 5 7 1 2 9 4 6 8]
print(x[y])
[9 7 5 5 2 2 2 1 1 1]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

二维情形：

x = np.random.rand(5,5)*10
x = np.around(x,2)
print(x)
[[3.53 5.32 1.1  8.35 6.18]
 [9.09 8.01 4.69 4.28 6.88]
 [1.54 3.29 4.08 1.37 3.11]
 [0.   3.57 8.88 0.97 3.29]
 [7.59 8.89 3.98 2.4  1.73]]

y = np.argsort(x)
print(y)
[[2 0 1 4 3]
 [3 2 4 1 0]
 [3 0 4 1 2]
 [0 3 4 1 2]
 [4 3 2 0 1]]

y = np.argsort(x,axis = 0)#按列从小到大排序，输出对应索引
print(y)
[[3 2 0 3 4]
 [2 3 4 2 2]
 [0 0 2 4 3]
 [4 1 1 1 0]
 [1 4 3 0 1]]

y = np.argsort(x,axis = 1)#按行从小到大排序，输出对应索引
print(y)
[[2 0 1 4 3]
 [3 2 4 1 0]
 [3 0 4 1 2]
 [0 3 4 1 2]
 [4 3 2 0 1]]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

应用示例：
取数组x中最小的三个值：

x = np.array([2,6,8,4,3,7,9,1])
y = np.argsort(x)
z = y[0:3]
print(x[z])
[1 2 3]
1
2
3
4
5