如何在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)