如何获得列表中所有列表组合的总和？

Question

我有一个列表列表，并试图获得所有组合的总和。

例如： [[1,2,3],[2,3],[4,5]]将产生(1+2+4),(1+2+5),(1+3+4),(1+3+5), (2+2+4)等

我似乎找不到办法做到这一点 - 有什么帮助吗？ 请注意，我知道如何产生总和，问题只是迭代所有组合。

我有使用 zip() 的想法，但这会总结所有内容，然后我们可以旋转列表以涵盖所有组合，这可行吗？ 看起来很复杂。 或者有没有更简单的方法来使用循环进行迭代？

Answer 1

你可以使用itertools.product

lists = [[1, 2, 3], [2, 3], [4, 5]]
products = itertools.product(*lists)
for product in products:
    print(f'{product} {sum(product)}')

Output：

(1, 2, 4) 7
(1, 2, 5) 8
(1, 3, 4) 8
(1, 3, 5) 9
(2, 2, 4) 8
(2, 2, 5) 9
(2, 3, 4) 9
(2, 3, 5) 10
(3, 2, 4) 9
(3, 2, 5) 10
(3, 3, 4) 10
(3, 3, 5) 11

Answer 2

最简单的方法是使用itertools.combinations 。

将列表列表展平为一个列表，并通过生成器获取所有组合。

对每种组合做任何你想做的事情（总和）。

from itertools import combinations

a = [[1,2,3],[2,3],[4,5]]
flat_a = [item for sublist in a for item in sublist]
for combination in combinations(flat_a, 3):
    print(f"{combination}, {sum(combination)}")

参考：

展平
组合

Answer 3

from itertools import product
a = [[1,2,3],[2,3],[4,5]]
for i in product(a):
    print(i," ",sum(i))