元素添加两个元组

Question

我只是想知道是否有一种特别pythonic方式添加两个元组elementwise？

到目前为止（a和b元组），我有

map(sum, zip(a, b)) 

我的预期输出是：

(a[0] + b[0], a[1] + b[1], ...) 

和可能的体重将是给一个0.5的重量和b 0.5体重，等等。 （我试图采取加权平均）。

哪个工作正常，但说我想添加一个权重，我不太确定我该怎么做。

谢谢

Answer 1

拉他们，然后求和每个元组。

[sum(x) for x in zip(a,b)] 

编辑：这里有一个更好的，虽然更复杂的版本，它允许进行加权。

from itertools import starmap, islice, izip 

a = [1, 2, 3] 
b = [3, 4, 5] 
w = [0.5, 1.5] # weights => a*0.5 + b*1.5 

products = [m for m in starmap(lambda i,j:i*j, [y for x in zip(a,b) for y in zip(x,w)])] 

sums = [sum(x) for x in izip(*[islice(products, i, None, 2) for i in range(2)])] 

print sums # should be [5.0, 7.0, 9.0] 

Answer 2

>>> a = (1, 2, 3) 
>>> b = (4, 5, 6) 
>>> def averageWeightedSum(args): 
     return sum(args)/len(args) 
>>> tuple(map(averageWeightedSum, zip(a, b))) 
(2.5, 3.5, 4.5) 

另一种方法是首先应用的权重。这也将让你有不同的权重：

>>> from operator import mul 
>>> weights = (0.3, 0.7) 
>>> tuple(sum(map(mul, x, weights)) for x in zip(a, b)) 
(3.0999999999999996, 4.1, 5.1) 
>>> weights = (0.5, 0.5) 
>>> tuple(sum(map(mul, x, weights)) for x in zip(a, b)) 
(2.5, 3.5, 4.5) 

Answer 3

就拿公式一对坐标的加权和，并与每对（注意for后两个变量）是迭代器的元组：

tuple(0.5*an + 0.5*bn for an, bn in zip(a, b)) 

这可以使它简单易读并且可以作为一行代码。当然，如果你的“加权和”是一个复杂的函数，你可以先将它定义为一个单独的函数。

Answer 4

如果你不介意的依赖，可以在阵列

使用numpy的进行的elementwise操作

>>> import numpy as np 
>>> a = np.array([1, 2, 3]) 
>>> b = np.array([3, 4, 5]) 
>>> a + b 
array([4, 6, 8]) 

Answer 5

如果你的元组包含str对象：

list(map(''.join, zip('abc', '123'))) 
# Returns ['a1', 'b2', 'c3']