sort方法使用,对于多维数据可以指定对指定轴排序,默认是最里面一层数据排序
In [8]: arr = np.random.randn(3,2,2)
In [9]: arr
Out[9]:
array([[[ 1.17120187, 1.46243415],
[ 1.03303677, 0.046324 ]],
[[ 0.54480189, -1.53867658],
[-2.71762905, -0.8470151 ]],
[[-0.8524279 , -1.50170625],
[-0.32525088, 0.33931581]]])
In [10]: arr.sort()#等价于arr.sort(axis = 2) 默认对最内层排序
In [11]: arr
Out[11]:
array([[[ 1.17120187, 1.46243415],
[ 0.046324 , 1.03303677]],
[[-1.53867658, 0.54480189],
[-2.71762905, -0.8470151 ]],
[[-1.50170625, -0.8524279 ],
[-0.32525088, 0.33931581]]])
In [12]: arr.sort(axis = 1)#对第二维的数据排序
In [13]: arr
Out[13]:
array([[[ 0.046324 , 1.03303677],
[ 1.17120187, 1.46243415]],
[[-2.71762905, -0.8470151 ],
[-1.53867658, 0.54480189]],
[[-1.50170625, -0.8524279 ],
[-0.32525088, 0.33931581]]])