Sum of list of lists; returns sum list

by

You could try this:

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]

This uses a combination of zip and * to unpack the list and then zip the items according to their index. You then use a list comprehension to iterate through the groups of similar indices, summing them and returning in their ‘original’ position.

To hopefully make it a bit more clear, here is what happens when you iterate through zip(*l):

In [13]: for i in zip(*l):
   ....:     print i
   ....:     
   ....:     
(3, 1, 9)
(7, 4, 8)
(2, 5, 7)

In the case of lists that are of unequal length, you can use itertools.izip_longest with a fillvalue of 0 – this basically fills missing indices with 0, allowing you to sum all ‘columns’:

In [1]: import itertools

In [2]: l = [[3,7,2],[1,4],[9,8,7,10]]

In [3]: [sum(i) for i in itertools.izip_longest(*l, fillvalue=0)]
Out[3]: [13, 19, 9, 10]

In this case, here is what iterating over izip_longest would look like:

In [4]: for i in itertools.izip_longest(*l, fillvalue=0):
   ...:     print i
   ...:     
(3, 1, 9)
(7, 4, 8)
(2, 0, 7)
(0, 0, 10)

More Related Contents:

  1. Find the sum of two lists of lists element-wise [closed]
  2. Transpose/Unzip Function (inverse of zip)?
  3. How to find the cumulative sum of numbers in a list?
  4. Sum a list of numbers in Python
  5. Add SUM of values of two LISTS into new LIST
  6. Summing elements in a list
  7. How to input matrix (2D list) in Python?
  8. Co-occurrence matrix from nested list of words
  9. Generate all possible lists of length N that sum to S in Python
  10. Accessing the index in ‘for’ loops
  11. Finding the index of an item in a list
  12. What is the difference between Python’s list methods append and extend?
  13. Get a list of numbers as input from the user
  14. python list by value not by reference [duplicate]
  15. Create nice column output in python
  16. Comparing two lists using the greater than or less than operator
  17. Appending the same string to a list of strings in Python
  18. How to get first element in a list of tuples?
  19. Getting only element from a single-element list in Python?
  20. Removing a list of characters in string
  21. How to get last items of a list in Python?
  22. Generating sublists using multiplication ( * ) unexpected behavior [duplicate]
  23. Converting Dictionary to List? [duplicate]
  24. Appending to the same list from different processes using multiprocessing
  25. How to use python numpy.savetxt to write strings and float number to an ASCII file?
  26. Checking if any elements in one list are in another [duplicate]
  27. Block tridiagonal matrix python
  28. Print a list in reverse order with range()?
  29. Creating a list in Python with multiple copies of a given object in a single line
  30. Python : how to append new elements in a list of list?

Leave a Comment