argsort()函數

第一步：先定義一個array數據
import numpy as np
x=np.array([2,4,5,3,-10,1])

第二步：輸出結果：
y=np.argsort(x)
print(y)
輸出結果為：y=[4 5 0 3 1 2]

第三步：總結：
argsort()函數是將x中的元素從小到大排列，提取其對應的index(索引號)
例如：x[4]=-10最小，所以y[0]=4,
同理：x[2]=5最大，所以y[5]=2。

看以下官方案例：

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

 

————————————————
版權聲明：本文為CSDN博主「ITxiaoke」的原創文章，遵循CC 4.0 BY-SA版權協議，轉載請附上原文出處鏈接及本聲明。
原文鏈接：https://blog.csdn.net/u014745194/article/details/73496836