一、解释
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)
结果: