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 进行排序。