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

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

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

Params:

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

举例:

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

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

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

指定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)

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

发布了25 篇原创文章 · 获赞 14 · 访问量 1136

猜你喜欢

转载自blog.csdn.net/A_pinkpig/article/details/105344445