Python3:numpy模块中的argsort()函数

来源:互联网 发布:东莞plc编程培训机构 编辑:程序博客网 时间:2024/06/11 03:15

Python3:numpy模块中的argsort()函数


  • 转载请注明作者和出处:http://blog.csdn.net/u011475210
  • 操作系统:WINDOWS 10
  • 软件版本:python-3.6.2-amd64
  • 编  者:WordZzzz

  argsort函数是Numpy模块中的函数:

>>> import numpy>>> 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 specifiedby the `kind` keyword. It returns an array of indices of the same shape as`a` that index data along the given axis in sorted order.

  从中可以看出argsort函数返回的是数组值从小到大的索引值

Examples:

One dimensional array:一维数组

>>> x = np.array([3, 1, 2])>>> np.argsort(x)array([1, 2, 0])

Two-dimensional array:二维数组

>>> x = np.array([[0, 3], [2, 2]])>>> xarray([[0, 3],[2, 2]])>>> np.argsort(x, axis=0) #按列排序array([[0, 1],[1, 0]])>>> np.argsort(x, axis=1) #按行排序array([[0, 1],[0, 1]])

Examples:

>>> x = np.array([3, 1, 2])>>> np.argsort(x) #按升序排列array([1, 2, 0])>>> np.argsort(-x) #按降序排列array([0, 2, 1])>>> x[np.argsort(x)] #通过索引值排序后的数组array([1, 2, 3])>>> x[np.argsort(-x)]array([3, 2, 1])

另一种方式实现按降序排序:

>>> a = x[np.argsort(x)]>>> aarray([1, 2, 3])>>> a[::-1]array([3, 2, 1]) 

系列教程持续发布中,欢迎订阅、关注、收藏、评论、点赞哦~~( ̄▽ ̄~)~

完的汪(∪。∪)。。。zzz

原创粉丝点击