python - python中嵌套元组的元素明智操作

Question

按元素添加元组非常简单:

a = (1,2,3,4)
b = (2,4,6,8)
tuple(x+y for x,y in zip(a,b))

(3,6,9,12)

但是嵌套元组的逐元素加法更复杂:

a = ((1,2),(3,4),(5,6))
b = ((2,4),(6,8),(10,12))
tuple(x+y for tup_a, tup_b in zip(a,b) for x,y in zip(tup_a,tup_b))

(3,6,9,12,15,18)

元组被展平了。如何在保留元组结构的同时对 nusted 元组执行逐元素加法？

这是期望的输出:

((3,6),(9,12),(15,18))

Answer 1

嵌套生成器表达式:

tuple(tuple(x + y for x, y in zip(tup_a, tup_b)) for tup_a, tup_b in zip(a, b))

演示:

>>> a = ((1,2),(3,4),(5,6))
>>> b = ((2,4),(6,8),(10,12))
>>> tuple(tuple(x + y for x, y in zip(tup_a, tup_b)) for tup_a, tup_b in zip(a, b))
((3, 6), (9, 12), (15, 18))