笔试问题第三

Python中对输入的二维/多维数组进行复合排序

如，输入的是

data = [[2,2,5],[2,1,3],[1,2,3],[3,1,4]]

1、要求将数组先按照第一列升序，第二列升序，第三列升序的方式排序

data.sort(key=lambda x:(x[0],x[1],x[2]))
print(data)

>>>[[1, 2, 3], [2, 1, 3], [2, 2, 5], [3, 1, 4]]

2、要求将数组按照第一列升序，第二列降序

data.sort(key=lambda x:(x[0],-1*x[1]))
print(data)

>>>>>> [[1, 2, 3], [2, 2, 5], [2, 1, 3], [3, 1, 4]]

备注：还有一种方法是调用numpy的函数lexsort，如：

>>> import numpy as np
>>> data = np.array([[2,2,5],[2,1,3],[1,2,3],[3,1,4]])
>>>> data
array([[2, 2, 5],
  [2, 1, 3],
  [1, 2, 3],
  [3, 1, 4]])

 将数组按照第一列降序，第二列升序，第三列升序的方式排序：

>>> idex=np.lexsort([data[:,2], data[:,1], -1*data[:,0]])
>>> sorted_data = data[idex, :]
>>> sorted_data
array([[3, 1, 4],
  [2, 1, 3],
  [2, 2, 5],
  [1, 2, 3]])