关于python中argsort()函数的使用

Posted LLLiuye

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了关于python中argsort()函数的使用相关的知识,希望对你有一定的参考价值。

在实现《机器学习实战》中kNN代码时遇到需要将计算好的距离进行排序,即可使用argsort()函数,在此依据个人理解对该函数进行简单的介绍。
总的来说,argsort()函数是对数组中的元素进行从小到大排序,并返回相应序列元素的数组下标。
以下通过例子进行详细解释。

1. 先定义一个数组

>>>from numpy import *
>>>a = array([7, 8, 5, -3, 10, 9])

2. 调用argsort()函数,将返回的值赋给y,并查看y的结果

>>>y = a.argsort()
>>>print(y)

结果为:y = array([3, 2, 0, 1, 5, 4])

按照数组a元素从小到大排序为:a[3]=-3,a[2]=5,a[0]=7,a[1]=8,a[5]=9,a[4]=10
而数组y保存的即为从小到大排序的数组a的下标。

3. 关于argsort()[num]的使用

num >= 0时,argsort()[num]的值即为y[num]的值
num < 0时,argsort()[num]的值为y数组反向输出的第num个数。

>>>x = a.argsort()[-1]

输出为:

4

即输出数组a中最大元素的下标,也是y数组中最后一个元素的值。

以上是关于关于python中argsort()函数的使用的主要内容,如果未能解决你的问题,请参考以下文章

np.argsort()函数,model.predict函数,python中reshape

np.argsort()函数,model.predict函数,python中reshape

python中argsort的使用

python中的反向排序和argsort

python 在机器学习中应用函数

python argsort()究竟如何返回的?