使用python做机器学习的一般都是从k-近邻算法开始学习。
这就很难绕开argsort()函数,但是在使用argsort()函数时大部分初学者都会被绕晕进去。
百度的答案大都来自同一篇文章的转载,并没有详细解释,这里详细解释一下。
argsort()函数是用来返回数组值从小到大索引值的。举例
a = [6,4,5]
那么
a.argsort() = [1,2,0]
下面讲一下详细逻辑:
我们把矩阵a从小到大排序,记排序后的矩阵为b:
b = [4,5,6]
矩阵a和b的关系:
b的第一个元素“4”对应的是a矩阵的第2个元素,也就是a[1]
b的第二个元素“5”对应的是a矩阵的第3个元素,也就是a[2]
b的第三个元素“6”对应的是a矩阵的第1个元素,也就是a[0]
所以a.argsort(),也就是a排序后索引值就是[1,2,0]
a.argsort() = [1,2,0]
PS:索引值相当于页数,是一个排序值,不等于元素的值。
像书的目录一样,对于例子中的矩阵,我们有如下索引:
6............0
4............1
5............2
所以排序后b=[4,5,6],用索引值来表达就是[1,2,0]