Tìm tối đa danh sách các danh sách theo tổng các phần tử trong Python

Question

Cách thành ngữ để làm maximumBy (hàm bậc cao hơn lấy hàm so sánh cho thử nghiệm), trên danh sách danh sách, nơi so sánh chúng tôi muốn thực hiện là tổng của danh sách, bằng Python?

Dưới đây là một Haskell thực hiện và dụ đầu ra:

> maximumBy (compare `on` sum) [[1,2,3],[4,5,6],[1,3,5]] 
> [4,5,6] 

Và triển khai những chức năng thư viện cơ sở, chỉ cho đầy đủ (trong trường hợp bạn muốn sử dụng làm giảm hoặc một cái gì đó :)

maximumBy cmp xs = foldl1 maxBy xs 
    where 
     maxBy x y = case cmp x y of GT -> x; _ -> y 

k `on` f = \x y -> f x `k` f y 

sum  = foldl' (+) 0 

Answer 1

Kể từ Python 2.5 bạn có thể sử dụng max với thông số chính:

>>> max(a, key=sum) 
[4, 5, 6] 

Answer 2

Nó không phải là terribly hiệu quả, nhưng:

reduce(lambda x,y: x if sum(x)>sum(y) else y, [[1,2,3],[4,5,6],[1,3,5]]) 

Answer 3

Nếu max không có thông số quan trọng bạn có thể mã mô hình DSU một cách rõ ràng:

max(izip(imap(sum,a),a))[1] 

izip và imap là từ các module itertools trong python 2 và làm những gì zip và bản đồ làm, nhưng lười biếng bằng cách sử dụng máy phát điện Python, để tránh consing một danh sách trung gian. Trong Python 3, bản đồ và zip được xây dựng là lười biếng.