Python_numpy.argsort函数

最新推荐文章于 2024-11-20 20:22:02 发布

Rocky Ding*

最新推荐文章于 2024-11-20 20:22:02 发布

阅读量265

点赞数

CC 4.0 BY-SA版权

分类专栏： Python 文章标签： python numpy 索引深度学习机器学习

本文链接：https://blog.csdn.net/Rocky6688/article/details/107847340

Python 专栏收录该内容

50 篇文章

订阅专栏

本文详细介绍了 NumPy 中的 argsort 函数，通过实例展示了如何使用此函数获取数组元素排序后的索引，包括一维和二维数组的排序方式，并区分了按行和按列的排序方法。

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

该函数将会返回数组排序后的对应索引（从小到大）。

我们直接看具体例子：

import numpy as np

    x = np.array([3,1,2,5,3])
    print(np.argsort(x))

    x2 = np.array([[0,1,3], [3,2,5]])
    print(np.argsort(x2)) # 默认是左右比较

    print((np.argsort(x2, axis=0))) # 上下比较
    print((np.argsort(x2, axis=1))) # 左右比较

--------------结果如下---------------

[1 2 0 4 3]

[[0 1 2]
 [1 0 2]]
 
[[0 0 0]
 [1 1 1]]
 
[[0 1 2]
 [1 0 2]]