绕晕大多数Python初学者的argsort()函数

绕晕大多数Python初学者的argsort()函数

转载 https://www.jianshu.com/p/a33593a83233
使用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]