【Numpy】sort、argsort、lexsort、partition、sorted

最新推荐文章于 2025-06-29 17:24:25 发布

转载最新推荐文章于 2025-06-29 17:24:25 发布 · 350 阅读

1 ·

CC 4.0 BY-SA版权

原文链接：https://blog.youkuaiyun.com/haiyang_duan/article/details/79221458#commentBox

Python 专栏收录该内容

368 篇文章

订阅专栏

本文深入讲解了NumPy库中排序函数argsort的使用方法，包括一维和二维数组的升序、降序排序，以及如何根据特定列或行对数组进行整体排序。通过实例演示了argsort函数返回索引值的特点，帮助读者理解并掌握该函数的应用。

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

排序（sort、argsort、lexsort、partition、sorted）参数讲解

参考博客：https://blog.youkuaiyun.com/haiyang_duan/article/details/79221458#commentBox

argsort函数

>>> 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 specified
by 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]])
>>> x
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]]

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)]
>>> a
array([1, 2, 3])
>>> a[::-1]
array([3, 2, 1])

使用argsort实现按照某列对数组进行整体排序

按照某列排序

data = data[data[:,2].argsort()] #按照第3列对行排序

注意：argsort返回的只是排好序后的行索引，不会改变原数组。

按照某行进行排序，可以利用转置操作，代码如下所示：

data = data.T(data.T[:,2].argsort()).T # 按照第3行对列进行排序

也可以直接按行进行排序，代码如下：

data = data[:,data[2].argsort()]