Numpy之argsort函数用法

我就是我 2022-06-10 04:29 275阅读 0赞

文档原文：  
[https://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html][https_docs.scipy.org_doc_numpy_reference_generated_numpy.argsort.html]

--------------------

## 函数用法： ##

**numpy.argsort(a, axis=-1, kind=’quicksort’, order=None)\[source\]**

> argsort函数返回的是数组值从小到大的索引值

## 举例如下： ##

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

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

关于相关排序的还有两个常用的函数sort和sorted，详细内容请看：[http://maoersong.blog.163.com/blog/static/171557351201424105925681/?newFollowBlog][http_maoersong.blog.163.com_blog_static_171557351201424105925681_newFollowBlog]

[https_docs.scipy.org_doc_numpy_reference_generated_numpy.argsort.html]: https://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html
[http_maoersong.blog.163.com_blog_static_171557351201424105925681_newFollowBlog]: http://maoersong.blog.163.com/blog/static/171557351201424105925681/?newFollowBlog