Python NumPy 数组 排序

前言

NumPy（Numerical Python的缩写）是一个开源的Python科学计算库。使用NumPy，就可以很自然地使用数组和矩阵。NumPy包含很多实用的数学函数，涵盖线性代数运算、傅里叶变换和随机数生成等功能。本文主要介绍Python NumPy 数组排序。

1、数组排序

NumPy ndarray对象具有一个名为sort()的函数，该函数将对指定的数组进行排序。

例如：

排序数组：

import numpy as np

arr = np.array([103, 211, 120, 911])

print(np.sort(arr))

上面代码运行结果为：

[103 120 211 911]

注意：此方法返回数组的副本，而原始数组保持不变。

还可以对字符串数组或任何其他数据类型进行排序：

例如：

按字母顺序对数组排序：

import numpy as np

arr = np.array(['ban', 'cpu', 'app'])

print(np.sort(arr))

上面代码运行结果为：

['app' 'ban' 'cpu']

例如：

对布尔数组进行排序：

import numpy as np

arr = np.array([True, False, True])

print(np.sort(arr))

上面代码运行结果为：

[False  True  True]

2、排序 2-D Array

如果在二维数组上使用sort()方法，则两个数组都将被排序：

例如：

排序二维数组：

import numpy as np

arr = np.array([[3, 12, 4], [5, 10, 1]])
arr1 = np.array([[3, 12, 4, 5, 10, 1]])

print(np.sort(arr))
print(np.sort(arr1))

上面代码运行结果为：

[[ 3  4 12]
 [ 1  5 10]]
[[ 1  3  4  5 10 12]]