Python: Sum two or more lists, the lengths of the lists may be different

Python List: Exercise - 194 with Solution

Write a Python program to sum two or more lists, the lengths of the lists may be different.

Sample Solution:

Python Code:

def sum_lists_diff_length(test_list):
    result =  [sum(x) for x in zip(*map(lambda x:x+[0]*max(map(len, test_list)) if len(x)<max(map(len, test_list)) else x, test_list))]
    return result

nums = [[1,2,4],[2,4,4],[1,2]]
print("\nOriginal list:")
print(nums)
print("Sum said lists with different lengths:")
print(sum_lists_diff_length(nums))

nums = [[1],[2,4,4],[1,2],[4]]
print("\nOriginal list:")
print(nums)
print("Sum said lists with different lengths:")
print(sum_lists_diff_length(nums))

Sample Output:

Original list:
[[1, 2, 4], [2, 4, 4], [1, 2]]
Sum said lists with different lengths:
[4, 8, 8]

Original list:
[[1], [2, 4, 4], [1, 2], [4]]
Sum said lists with different lengths:
[8, 6, 4]

Pictorial Presentation:

Flowchart:

Visualize Python code execution:

The following tool visualize what the computer is doing step-by-step as it executes the said program:

Python Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a Python program to find the dimension of a given matrix.
Next: Write a Python program to traverse a given list in reverse order, also print the elements with original index.