Python对二维数组按指定列排序

import numpy as np
import numpy as np
data = np.array([[1,9,3,4,5], [1,3,3,6,7], [2,2,4,5,7], [3,1,5,6,7], [4,6,6,7,8]])
# 按照第一列排序
idex=np.lexsort([data[:,1]])
sorted_data = data[idex, :]
print(sorted_data)
# 按照第二列排序
idex=np.lexsort([data[:,1]])
sorted_data = data[idex, :]
print(sorted_data)