在嵌套列表中查找列表具有最大的值总和

Question

我有嵌套列表，我试图找出更好的方法来查找列表具有最大值的总和。以下是我的嵌套列表：nums = [[[1, 3000], [1, 2000]],        [[1, 3000], [2, 3000]],        [[1, 3000], [3, 4000]],        [[2, 5000], [1, 2000]],        [[2, 5000], [2, 3000]],        [[2, 5000], [3, 4000]],        [[3, 4000], [1, 2000]],        [[3, 4000], [2, 3000]],        [[3, 4000], [3, 4000]]]所需的输出 =[[2, 5000], [3, 4000]]因为值的总和最大。我的做法：largest = []for i in range(len(nums)-1):  if (nums[i][0][1] + nums[i][1][1]) > (nums[i+1][0][1] + nums[i+1][1][1]):    largest.append(nums[i])print(largest)

Answer 1

这是使用max和自定义函数的一种方法：

from operator import itemgetter

res = max(nums, key=lambda x: sum(map(itemgetter(1), x)))

[[2, 5000], [3, 4000]]

Answer 2

DATA = [

    [[1, 3000], [1, 2000]],

    [[1, 3000], [2, 3000]],

    [[1, 3000], [3, 4000]],

    [[2, 5000], [1, 2000]],

    [[2, 5000], [2, 3000]],

    [[2, 5000], [3, 4000]],

    [[3, 4000], [1, 2000]],

    [[3, 4000], [2, 3000]],

    [[3, 4000], [3, 4000]],

]

tups = [

    (sum(n for _, n in row), row)

    for row in DATA

]

mx = max(tups)

print(mx)                        # (9000, [[2, 5000], [3, 4000]])

print([xs[0] for xs in mx[1]])   # [2, 3]

Answer 3

使用总和作为键对元素进行排序：

max(DATA, key=lambda x:x[0][1] + x[1][1])

#[[2, 5000], [3, 4000]]

这是发布数据的最快解决方案。