如何在python中按最大两个轴对列表进行排序？

Question

我想按列表中每个元素的最大索引0和索引1对列表进行排序。

例如，如果我有一个这样的列表：

[[2,1,2],
 [0,3,9],
 [1,2,9],
 [1,1,2],
 [3,2,7]]

排序后，它应如下所示：

[[0,3,9],
 [3,2,7],
 [2,1,2],
 [1,2,9],
 [1,1,2]]

那么什么是有效的方法呢？

我更喜欢使用numpy，但也欢迎使用其他解决方案。

Answer 1

取前两列的逐行最大值argsort，然后取反以得到降序。

您尚未指定任何决胜局，因此这与所需的输出不匹配。 换句话说，没有独特的解决方案可以解决您的问题。

res = A[A[:, :2].max(1).argsort()[::-1]]

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

Answer 2

只需使用带有自定义键功能的sort ：

data = [[2,1,2],[0,3,9],[1,2,9],[1,1,2],[3,2,7]]


data.sort(key=lambda x: max(x[0:2]),reverse=True)