Python Numpy学习（二）排序

一、简单排序

import numpy as np

a_array = np.array([[1.5,1.3,7.5],
                  [5.6,7.8,1.2]])
a_array

array([[1.5, 1.3, 7.5],
       [5.6, 7.8, 1.2]])

按行排序

np.sort(a_array)

array([[1.3, 1.5, 7.5],
       [1.2, 5.6, 7.8]])

按列排序

np.sort(a_array, axis = 0)

array([[1.3, 1.5, 7.5],
       [1.2, 5.6, 7.8]])

输出排序后的索引变化

np.argsort(a_array)

array([[1, 0, 2],
       [2, 0, 1]])

二、等分与插入数据

b_array = np.linspace(0,10,10)
b_array

array([ 0.        ,  1.11111111,  2.22222222,  3.33333333,  4.44444444,
        5.55555556,  6.66666667,  7.77777778,  8.88888889, 10.        ])

values =np.array([2.5,6.5,9.5])

将values的值插入到数组中，得到插入位置的索引（排序号的数组才可以用）

np.searchsorted(b_array,values)

array([3, 6, 9])

三、分列排序

c_array = np.array([[1,0,6],[1,7,0],[2,3,1],[2,4,0]])
c_array

array([[1, 0, 6],
       [1, 7, 0],
       [2, 3, 1],
       [2, 4, 0]])

index = np.lexsort([-1*c_array[:,0], c_array[:,2]])
index

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

d_array = c_array[index]
d_array

array([[2, 4, 0],
       [1, 7, 0],
       [2, 3, 1],
       [1, 0, 6]])