【发布时间】:2013-11-14 19:13:01
【问题描述】:
我有这个元组列表:
a = [(1, 2), (1, 4), (1, 6)]
我想使用reduce函数来得到这个结果:
(3, 12)
我试过了:
x = reduce(lambda x, y: x+y, a)
但我得到一个错误...我想将每个元组的第一个索引中的所有元素相加,然后将第二个元素相加。
【问题讨论】:
我有这个元组列表:
a = [(1, 2), (1, 4), (1, 6)]
我想使用reduce函数来得到这个结果:
(3, 12)
我试过了:
x = reduce(lambda x, y: x+y, a)
但我得到一个错误...我想将每个元组的第一个索引中的所有元素相加,然后将第二个元素相加。
【问题讨论】:
如果您希望reduce 的输出为元组,则所有中间结果也应为元组。
a = [(1, 2), (1, 4), (1, 6)]
print reduce(lambda x, y: (x[0] + y[0], x[1] + y[1]), a)
输出
(3, 12)
编辑:如果您想在列表为空时获取(0, 0)
a = []
print reduce(lambda x, y: (x[0] + y[0], x[1] + y[1]), [(0, 0)] + a)
输出
(0, 0)
编辑 2: Reduce 接受 default initializer 作为最后一个参数,这是可选的。通过使用它,代码变成了
a = []
print reduce(lambda x, y: (x[0] + y[0], x[1] + y[1]), a, (0, 0))
【讨论】:
reduce 接受可选的initializer: reduce(lambda x, y: (x[0] + y[0], x[1] + y[1]), a, (0, 0))
>>> a = [(1, 2), (1, 4), (1, 6)]
>>> map(sum, zip(*a))
[3, 12]
更新
根据雷蒙德·赫廷格的说法,
zip-star 技巧滥用堆栈来昂贵地计算转置。
这是一个不使用列表理解的替代方法。
>>> a = [(1, 2), (1, 4), (1, 6)]
>>> [sum(item[i] for item in a) for i in range(2)] # 2 == len(a[0])
[3, 12]
>>> a = []
>>> [sum(item[i] for item in a) for i in range(2)] # 2 == len(a[0])
[0, 0]
【讨论】:
tuple(map(sum, zip(*a)))
试试这个:
>>> from numpy import asarray
>>> a = asarray([(1, 2), (1, 4), (1, 6)])
>>> reduce(lambda x,y:x+y, a)
array([ 3, 12])
【讨论】:
你很亲密。只需修改您的代码以首先解压缩输入元组。添加新值后,只需重新打包结果元组:
>>> a = [(1, 2), (1, 4), (1, 6)]
>>> reduce(lambda (sx, sy), (x, y): (sx+x, sy+y), a)
(3, 12)
【讨论】: