Numpy之argsort函数用法

文档原文:
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

你可能感兴趣的:(Python学习笔记,机器学习学习笔记)