numpy中的argsort()

最新推荐文章于 2024-11-16 23:12:02 发布

原创最新推荐文章于 2024-11-16 23:12:02 发布 · 5.8k 阅读

6 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

20 篇文章

订阅专栏

本文详细介绍了 NumPy 中的 argsort 函数，该函数返回数组元素从小到大的索引值，适用于一维和二维数组。文章通过多个示例展示了如何使用 argsort 对数组进行升序和降序排序，并解释了 axis 参数的作用。

argsort()

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])