Dodawanie dwóch krotek elementwise

Question

Zastanawiam się, czy był szczególnie pythonic sposób dodawania dwóch krotek elementwise?

tej pory (a i b są krotki), mam

map(sum, zip(a, b)) 

Moje oczekiwany wynik będzie:

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

i ewentualnego ważenia byłoby dać 0,5 wagę i b 0.5 waga lub tak dalej. (Próbuję wziąć średnią ważoną).

Co działa dobrze, ale powiedz, że chciałem dodać wagę, nie jestem do końca pewien, jak bym to zrobił.

Dzięki

Answer 1

Zapnij je, a następnie zsumuj każdą krotkę.

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

EDIT: Oto lepiej, aczkolwiek bardziej skomplikowanej wersji, która pozwala na ważenie.

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) 

Alternatywą byłoby zastosowanie wagi pierwszy. To także pozwala mieć różne ciężary:

>>> 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

przybrać formułę ważonej sumy jednej pary współrzędnych i forma krotki z iteratora nad każdą parą (uwaga na dwie zmienne zachodzących po for):

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

To sprawia, że ​​jest on prosty i czytelny jako jednolinijkowy. Oczywiście, jeśli twoja "ważona suma" jest skomplikowaną funkcją, musisz najpierw zdefiniować ją jako osobną funkcję.

Answer 4

Jeśli nie przeszkadza zależność można użyć numpy dla operacji elementwise na tablicach

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

Answer 5

Jeśli krotki zawierać str obiekty:

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