Python NumPy sort方法用法及代码示例

例子

对一维数组进行排序

x = np.array([1,4,2,3])
np.sort(x)



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

对二维数组进行排序

假设我们有以下二维数组：

x = np.array([[1,4],[3,2]])
x



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

对展平的数组进行排序

np.sort(x, axis=None)



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

按行排序

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



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

按列排序

np.sort(x, axis=0)



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

指定结构化数组的顺序

假设我们有以下结构化数组：

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.sort(x, order="name")



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

请注意元组如何按名称排序： Alex 、 Bob 和 Cathy 。

按 age 排序：

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



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

请注意元组如何按年龄排序：15、24，然后是 35。

您还可以传入要排序的标签数组(例如 order=["name", "age"] )，这将首先按 name 对数组进行排序，然后按 age 进行排序。