对python中嵌套元组的元素操作

时间:2013-10-19 19:56:27

标签: python tuples

元素的元素添加相当简单:

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)

元组被夷为平地。如何在保留元组结构的同时对生成的元组执行元素加法?

这是所需的输出:

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

2 个答案:

答案 0 :(得分: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))

答案 1 :(得分:1)

def xsum(a,b):
    if isinstance(a,(list,tuple)) and isinstance(b,(list,tuple)):
       return [xsum(x,y) for x,y in zip(a,b)]
    return a+b

a = ((1,2),(3,4),(5,6))
b = ((2,4),(6,8),(10,12))
print [xsum(x,y) for x,y in zip(a,b)]

可能就是我要做的,这应该适用于更深层次的深度而不仅仅是两个