函数解析：argsort()函数

一、解释

argsort()函数是NumPy库中的一个函数，用于返回数组元素排序后的索引值。它可以用于一维或多维数组。

原型： numpy.argsort(a, axis=-1, kind='quicksort', order=None)

参数：

• a：待排序的数组。
• axis：指定排序的轴，默认为最后一个轴。
• kind：排序算法的种类，默认为’quicksort’（快速排序）。
• order：如果数组是结构化数组，则指定按照哪些字段进行排序。

TIPS：（个人觉得直接看应用会比较好）

二、应用

一维数组

1.直接使用

代码：

import numpy as np
#定义数组
x=np.array([2,6,3,8,9,-1])
#直接使用的两种形式
y=x.argsort()
z=np.argsort(x)
 
#打印
print(x)
print(y)
print(z)

结果：

可知，输出结果是将x数组从小到大排序后，返回的索引。

如果要显示排序后的数组，插入代码

z = x[y]

2.argsort()[num]的使用

代码：

import numpy as np
#定义数组
x=np.array([2,6,3,8,9,-1])
print(x)
 
#直接使用
y=x.argsort()
print(y)
 
#argsort()[num]
y=x.argsort()[-1]
print(y)

 结果：

 

 y=x.argsort()[-1]其实就是y[-1]

2.二维数组

代码：

import numpy as np
 
arr = np.array([[5, 2, 3],
[4, 1, 6],
[7, 8, 9]])
#索引数组
ndx = np.argsort(arr,axis=1)
#排序后的数组
sorted_arr = np.take_along_axis(arr, ndx, axis=1)#axis=1是按行排序，axis=0是按列排序
 
#打印
print(ndx)
print(sorted_arr)

结果：