Numpy 排序 -- sort()、argsort()

最新推荐文章于 2025-06-22 19:39:31 发布

Wang_Jiankun

最新推荐文章于 2025-06-22 19:39:31 发布

阅读量3.5k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Python 数据分析与展示文章标签： numpy 排序 sort() argsort()

本文链接：https://blog.youkuaiyun.com/Wang_Jiankun/article/details/83412800

Python 数据分析与展示专栏收录该内容

8 篇文章

订阅专栏

本文详细介绍了Numpy中的排序函数sort()和argsort()的使用方法，包括参数解释、排序算法选择及多级排序操作，通过实例展示了如何对一维和二维数组进行内容排序和下标排序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Numpy 排序 – sort()、argsort()

官方文档

1、内容排序：sort()

函数原型： 对数组本身进行排序

numpy.sort(a, axis=-1, kind='quicksort', order=None)

参数：

a：要排序的 numpy 数组
axis：排序的维度，默认为-1最后一维，None为所有元素拉平后排序
kind：排序算法，有：‘quicksort’, ‘mergesort’, ‘heapsort’, ‘stable’
order：多级排序的顺序，需要构建一个field

实例：

import numpy as np
a = np.array([[1,4],[3,1]])

# 默认按最后一维排序，每行各自排序
np.sort(a)               
=> narray([[1, 4],
           [1, 3]])

# 按地1维排序 ，每列各自排序         
np.sort(a, axis=0)
=> narray([[1, 1],
           [3, 4]])

# 所有元素排序
np.sort(a, axis=None)
=> narray([1, 1, 3, 4])

# 多级排序
dtype = [('name', 'S10'), ('height', float), ('age', int)]
values = [('Arthur', 1.8, 41),  ('Lancelot', 1.9, 38), ('Galahad', 1.7, 38)]
a = np.array(values, dtype=dtype)
np.sort(a, order='height')                        
=> array([('Galahad', 1.7, 38), ('Arthur', 1.8, 41), ('Lancelot', 1.8999999999999999, 38)],
      	 dtype=[('name', '|S10'), ('height', '<f8'), ('age', '<i4')])

2、下标排序：argsort()

函数原型： 数组本身不变，返回排序的下标数组

numpy.argsort(a, axis=-1, kind='quicksort', order=None)

参数：

a：要排序的 numpy 数组
axis：排序的维度，默认为-1最后一维，None为所有元素拉平后排序
kind：排序算法，有：‘quicksort’, ‘mergesort’, ‘heapsort’, ‘stable’
order：多级排序的顺序，需要构建一个field

实例：

import numpy as np

# 一维数组
x = np.array([3, 1, 2])
np.argsort(x)
=> array([1, 2, 0])

# 二维数组
x = np.array([[0, 3], [2, 2]])

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

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