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

使用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]

    原文作者:python
    原文地址: https://www.jianshu.com/p/a33593a83233
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞