Python numpy argsort用法及代碼示例

本文簡要介紹 python 語言中 numpy.argsort 的用法。

用法:

numpy.argsort(a, axis=- 1, kind=None, order=None)

返回將對數組進行排序的索引。

使用 kind 關鍵字指定的算法沿給定軸執行間接排序。它返回一個索引數組，其形狀與索引數據沿給定軸按排序順序排列。

參數：

a： array_like

要排序的數組。

axis： int 或無，可選

要排序的軸。默認值為 -1(最後一個軸)。如果為 None，則使用展平數組。

kind： {‘quicksort’, ‘mergesort’, ‘heapsort’, ‘stable’}，可選

排序算法。默認值為‘quicksort’。請注意，‘stable’ 和 ‘mergesort’ 都在後台使用 timsort，通常，實際實現會因數據類型而異。保留 ‘mergesort’ 選項是為了向後兼容。

order： str 或 str 列表，可選

當 a 是定義了字段的數組時，此參數指定首先比較哪些字段，第二個等。單個字段可以指定為字符串，不需要指定所有字段，但仍會使用未指定的字段，在他們在 dtype 中出現的順序，以打破關係。

返回：

index_array： ndarray，int

排序的索引數組a沿著指定的軸.如果a是一維的，a[index_array]產生一個排序的a.更普遍，np.take_along_axis(a, index_array, axis=axis)總是產生排序的a，與維度無關。

注意：

有關不同排序算法的說明，請參閱 sort 。

從 NumPy 1.4.0 開始，argsort 適用於包含 nan 值的實數/複數數組。增強的排序順序記錄在 sort 中。

例子：

一維數組：

>>> x = np.array([3, 1, 2])
>>> np.argsort(x)
array([1, 2, 0])

二維數組：

>>> x = np.array([[0, 3], [2, 2]])
>>> x
array([[0, 3],
       [2, 2]])

>>> ind = np.argsort(x, axis=0)  # sorts along first axis (down)
>>> ind
array([[0, 1],
       [1, 0]])
>>> np.take_along_axis(x, ind, axis=0)  # same as np.sort(x, axis=0)
array([[0, 2],
       [2, 3]])

>>> ind = np.argsort(x, axis=1)  # sorts along last axis (across)
>>> ind
array([[0, 1],
       [0, 1]])
>>> np.take_along_axis(x, ind, axis=1)  # same as np.sort(x, axis=1)
array([[0, 3],
       [2, 2]])

N 維數組的已排序元素的索引：

>>> ind = np.unravel_index(np.argsort(x, axis=None), x.shape)
>>> ind
(array([0, 1, 1, 0]), array([0, 0, 1, 1]))
>>> x[ind]  # same as np.sort(x, axis=None)
array([0, 2, 2, 3])

用鍵排序：

>>> x = np.array([(1, 0), (0, 1)], dtype=[('x', '<i4'), ('y', '<i4')])
>>> x
array([(1, 0), (0, 1)],
      dtype=[('x', '<i4'), ('y', '<i4')])

>>> np.argsort(x, order=('x','y'))
array([1, 0])

>>> np.argsort(x, order=('y','x'))
array([0, 1])