Python NumPy argsort方法用法及代碼示例

例子

對一維數組進行排序

x = np.array([7,5,8,6])
np.sort(x)



array([1, 3, 0, 2])

對二維數組進行排序

假設我們有以下二維數組：

x = np.array([[7,5],[8,6]])
x



array([[7, 5],
       [8, 6]])

對展平的數組進行排序

np.argsort(x, axis=None)



array([1, 3, 0, 2])

按行排序

np.argsort(x)      # or explicitly set axis=1



array([[1, 0],
       [1, 0]])

按列排序

np.argsort(x, axis=0)



array([[0, 0],
       [1, 1]])

指定結構化數組的順序

假設我們有以下結構化數組：

dtype = [('name', 'S15'), ('age', int)]
values = [('Bob', 35), ('Alex', 15),('Cathy', 24)]
x = np.array(values, dtype)
x



array([(b'Bob', 35), (b'Alex', 15), (b'Cathy', 24)],
        dtype=[('name', 'S15'), ('age', '<i8')])

我們的數組 x 由三個元組組成，第一個元素為 name ，第二個元素為 age 。

按 name 排序：

np.argsort(x, order="name")



array([1, 0, 2])

這裏返回的數組包含元組的原始索引 Alex 、 Bob 和 Cathy 。

按 age 排序：

np.argsort(x, order="age")



array([1, 2, 0])

這裏返回的數組包含年齡為 15、24 和 35 的元組的原始索引。

您還可以傳入要排序的標簽數組(例如 order=["name", "age"] )，這將首先按 name 對數組進行排序，然後按 age 進行排序。