关于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数组中最后一个元素的值。

转载于:https://www.cnblogs.com/lliuye/p/8629257.html