浅析python中numpy包中的argsort函数的使用
时间：2020-12-09 20:47:44 | 栏目：Python代码 | 点击：次
概述
argsort()函数在模块numpy.core.fromnumeric中。
在python中排序数组，或者获取排序顺序的时候，我们常常使用numpy包的argsort函数来完成。
如下图所示，是使用python获取到数组中的排序的顺序。
data=numpy.array([1,2,3,4,5]) datasort=numpy.argsort(data) datasort Out[39]: array([0, 1, 2, 3, 4], dtype=int64) data Out[40]: array([1, 2, 3, 4, 5]) datasort1=data.argsort() datasort1 Out[42]: array([0, 1, 2, 3, 4], dtype=int64)
我们也可以通过help(numpy.argsort)来查看使用方法
help(numpy.argsort) Help on function argsort in module numpy.core.fromnumeric: argsort(a, axis=-1, kind='quicksort', order=None) Returns the indices that would sort an array. Perform an indirect sort along the given axis using the algorithm specified by the `kind` keyword. It returns an array of indices of the same shape as
如果想要通过argsort实现排序可以使用切片实现
data1=numpy.array([1,3,4,56,2,0]) datasort=data1[data1.argsort()] datasort Out[48]: array([ 0, 1, 2, 3, 4, 56])
PS：NumPy 中argsort函数
排序函数，返回array类型
argsort函数返回的是数组值从小到大的元素的索引值
#!/usr/bin/env python # -*- coding: utf-8 -*- import numpy as np inX = np.array([1,2,-1,3,4,7,8]) print inX print inX.argsort()
总结