我正在尝试找到一种方法来组合 2 个列表,然后以特定方式对组合列表求和。前 2 个元素的总和必须与下一个元素相加。
例如:第一个组合 ((1, 2), (4,)) 的总和应为 是 (3,7,6),因为 1+2=3、3+4=7、2+4=6。
然后并排打印:((1, 2), (4,)) (3,7,6) IF它(组合及其总和)不包含某个元素(不知何故已经插入)。
这是我的代码和示例:
a=[1,2,3]
b=[4,5,6]
import itertools
sets = [a,b];
ks = [2, 1,]
combinations = itertools.product(*[itertools.combinations(set, k) for set, k in zip(sets, ks)])
for combination in combinations:
print (combination)
最佳答案
直接说:
for i in itertools.product(*[itertools.combinations(set, k) for set, k in zip(sets, ks)]):
i_sum = i[0][0] + i[0][1]
print(i, (i_sum, i_sum + i[1][0], i[0][1] + i[1][0]))
输出:
((1, 2), (4,)) (3, 7, 6)
((1, 2), (5,)) (3, 8, 7)
((1, 2), (6,)) (3, 9, 8)
((1, 3), (4,)) (4, 8, 7)
((1, 3), (5,)) (4, 9, 8)
((1, 3), (6,)) (4, 10, 9)
((2, 3), (4,)) (5, 9, 7)
((2, 3), (5,)) (5, 10, 8)
((2, 3), (6,)) (5, 11, 9)
关于python - (python) 我如何以特定方式对组合的所有结果求和?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46613380/