numpy中实用但不常见的方法(3)np.argsort

最新推荐文章于 2025-03-27 10:21:04 发布

陈华杰

最新推荐文章于 2025-03-27 10:21:04 发布

阅读量1.8w

点赞数 8

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python np.argsort

本文链接：https://blog.csdn.net/cetrol_chen/article/details/79148143

python 专栏收录该内容

10 篇文章

订阅专栏

本文详细介绍了 numpy 中的 argsort 函数，该函数能够返回数组中元素排序后的索引值。主要内容包括函数的基本用法、参数说明及示例代码。通过不同维度上的排序示例，帮助读者更好地理解如何使用此函数。

numpy.argsort(a, axis=-1, kind=’quicksort’, order=None)
功能: 将矩阵a按照axis排序，并返回排序后的下标
参数: a:输入矩阵， axis:需要排序的维度
返回值: 输出排序后的下标

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

>>> x = np.array([[1, 5, 7], [3, 2, 4]])
>>> np.argsort(x, axis=0)
array([[0, 1, 1],
       [1, 0, 0]])  #沿着行向下(每列)的元素进行排序
>>> np.argsort(x, axis=1)
array([[0, 1, 2],
       [1, 0, 2]])  #沿着列向右(每行)的元素进行排序

以上内容来自于官方API文档