Суммарный список списка кортежей

Question

Я обнаружил, что суммирование списка списка чисел:

In [9]: l = [[3,7,2],[1,4,5],[9,8,7]]
In [10]: [sum(i) for i in zip(*l)]
Out[10]: [13, 19, 14]

Однако я изо всех сил пытался найти способ суммировать список списка кортежей, чтобы привести к списку кортежей

In [9]: l = [[(1, 1), (1, 1), (1, 1)],[(2, 2), (2, 2), (2, 2)]]
In [10]: [(sum(i[0]), sum(i[1])) for i in zip(*l)] #Did not work

Это мой желаемый вывод:

Out[10]: [(3, 3), (3, 3), (3, 3)]

Спасибо!

Answer 1

Вы можете сначала рекурсивно сгладить итерацию

>>> import collections
>>> 
>>> def flatten(l):
...     for el in l:
...         if isinstance(el, collections.Iterable) and not isinstance(el, str):
...             for sub in flatten(el):
...                 yield sub
...         else:
...             yield el
... 
>>> sum(flatten(l))
18

См. эту ссылку для более подробной информации о flatten.

Answer 2

Вы можете использовать tuple понимание в пределах списка:

L = [[(1, 1), (1, 1), (1, 1)], [(2, 2), (2, 2), (2, 2)]]

res = [tuple(sum(j) for j in zip(*i)) for i in zip(*L)]

[(3, 3), (3, 3), (3, 3)]

Или использовать map для более функциональной альтернативы:

res = [tuple(map(sum, zip(*i))) for i in zip(*L)]

Для больших списков выможет пожелать рассмотреть стороннюю библиотеку, такую ​​как NumPy, где в терминологии NumPy вам нужна только сумма по указанной оси:

import numpy as np

A = np.array(L)

res = A.sum(axis=0)

array([[3, 3],
       [3, 3],
       [3, 3]])