数组排序

如果希望对NumPy数组中的元素进行排序，可以通过sort()方法实现，示例代码如下。

In [112]: arr = np.array([[6, 2, 7], [3, 6, 2], [4, 3, 2]])
In [113]: arr
Out[113]: 
array([[6, 2, 7],
​    [3, 6, 2],
​     [4, 3, 2]])
In [114]: arr.sort()
In [115]: arr
Out[115]: 
array([[2, 6, 7],
​     [2, 3, 6],
​     [2, 3, 4]])

从上述代码可以看出，当调用sort()方法后，数组arr中数据按行从小到大进行排序。需要注意的是，使用sort()方法排序会修改数组本身。

如果希望对任何一个轴上的元素进行排序，只需要将轴的编号作为sort()方法的参数传入即可。示例代码如下。

In [116]: arr = np.array([[6, 2, 7], [3, 6, 2], [4, 3, 2]])
In [117]: arr
Out[117]: 
array([[6, 2, 7],
​     [3, 6, 2],
​     [4, 3, 2]])
In [118]: arr.sort(0)    # 沿着编号为0的轴对元素排序
In [119]: arr
Out[119]: 
array([[3, 2, 2],
​     [4, 3, 2],
​     [6, 6, 7]])