在实现《机器学习实战》中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数组中最后一个元素的值。