码迷,mamicode.com
首页 > 编程语言 > 详细

关于python中argsort()函数的使用

时间:2018-03-23 13:03:51      阅读:527      评论:0      收藏:0      [点我收藏+]

标签:查看   距离   保存   img   简单   简单的   rom   输出   http   

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

关于python中argsort()函数的使用

标签:查看   距离   保存   img   简单   简单的   rom   输出   http   

原文地址:https://www.cnblogs.com/lliuye/p/8629257.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!