2.1.4 排序

NumPy提供了排序函数sort()来实现数组对象的排序。

NumPy的sort()函数与Python中列表的sort()函数功能类似，可以得到按值排序后的数组。稍微不同的是，NumPy中的sort()函数可以使用axis参数来指定在具体维度（轴）上进行排序。

在NumPy中，调用sort()函数的方式有以下两种。

（1）np.sort(Ndarray对象)。

（2）Ndarray对象.sort( )。

两种调用方式的排序结果相同，但是前一种方式会返回一个排列好的新数组，不会对原数组的顺序做修改。后一种调用方式会直接在原数组上进行重新排序。代码2-11展示了使用sort()函数排序。

代码2-11 使用sort()函数排序

import numpy as np
a = np.random.randint(-5, 6, size = (3, 4))
print("排序前的数组对象a：\n", a)
b = np.sort(a, axis = 1)   #对数组对象a按行排序
print("对数组对象a行排序后的结果：\n", b)

上述代码的输出结果为

排序前的数组对象a：

排序前的数组对象a：
 [[ 5 -3  0  2]
  [-4  2  4 -4]
  [-5  2 -3  2]]
对数组对象a行排序后的结果：
 [[-3   0  2  5]
  [-4  -4  2  4]
  [-5  -3  2  2]]