当前位置:   article > 正文

numpy 数组排序np.sort()、np.argsort()_np.sort()参数

np.sort()参数

np.sort(a, axis=-1, kind=‘quicksort’, order=None)

将指定轴上的每一个元素都按照从小到大的顺序排列。

Params:

  • axis:默认是-1,即最大维度,二维数组就是轴1,三维数组就是轴2。
  • kind:排序规则,有三个选项,如下,使用默认的就好
kindspeedworst casework spacestable
‘quicksort’1O(n^2)0no
‘mergesort’2O(n*log(n))~n/2yes
‘heapsort’3O(n*log(n))0no
  • order:指定排序的规则。

举例:

arr = np.array([[7,3,9],
                [9,0,8]])
np.sort(arr,axis=0)
array([[7, 0, 8],
       [9, 3, 9]])
  • 1
  • 2
  • 3
  • 4
  • 5

指定axis=0,把每一列按照从小到大排列。

np.sort(arr,axis=1)
array([[3, 7, 9],
       [0, 8, 9]])
  • 1
  • 2
  • 3

指定axis=1,把每一行按照从小到大排列。

np.argsort(a, axis=-1, kind=‘quicksort’, order=None)

返回的是数组值从小到大排序的索引值。就是把数组升序排序,然后看看各个元素在原来数组中的索引值,看看排在第一小的数在原来数组的哪个位置。形状与传入数组相同。
参数使用同上。
举例:
这里我们举一个简单的例子,万变不离其宗。

ind = np.argsort([4,3,5,6,0],axis=0)
ind
array([4, 1, 0, 2, 3], dtype=int64)
  • 1
  • 2
  • 3

附图解释。每个数字排序时,它的index也随着数值走。最后返回其原来的index。
在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/648871
推荐阅读
相关标签
  

闽ICP备14008679号