Python list is one of the very useful data types in Python. Python list is nothing but an array of elements. As we are interested to calculate the sum of the elements, we are considering all the elements in the list as numbers.
Problem Statement: Write a Python program to find the sum of all elements in the List.
Table of Contents
We can calculate the sum of all the elements in the Python list, using a simple building function sum().
[python]myList=[23,4,2,6,7]
print(sum(myList))[/python]
Output:
42
If you know the sum() function. This is a very simple and one-liner solution.
Many times, in interviews, you will ask to write your own Python program to calculate the sum of all the elements using recursion. You can not use any built-in function.
[python]
myList=[23,4,2,6,7]
def sumOfList(myList, nSum):
if len(myList):
return sumOfList(myList[1:], nSum+myList[0])
else:
return nSum
print(sumOfList(myList, 0))
[/python]
Output:
42
The code is self-explanatory. The only thing is that you should have a good understanding of how recursion work.
This problem was asked in McAfee coding round.
You can also solve it using for loop.
[python]
myList=[23,4,2,6,7]
def sumOfList(myList):
nSum=0
for i in myList:
nSum+=i
return nSum
print(sumOfList(myList))
[/python]
Output:
42
To use the reduce method you have to import functools module in your Python program.
Lambda is a special anonymous function. You can read more about the lambda function in Python.
[python]
import functools
myList=[23,4,2,6,7]
print(functools.reduce(lambda a, b: a+b, myList))
[/python]
Output:
42
Complexity:
This program has more complexity. As we are traversing each element and calling recursive function, it takes time O(n). As we are using recursion here, it requires extra memory to save the output from previous recursive calls.
This is a simple tutorial with a Python program to find the sum of all elements in the list. If you are beginners, I would recommend solving coding interview questions for practicing.
I am a Python enthusiast who loves Linux and Vim. I hold a Master of Computer Science degree from NIT Trichy and have 10 years of experience in the IT industry, focusing on the Software Development Lifecycle from Requirements Gathering, Design, Development to Deployment. I have worked at IBM, Ericsson, and NetApp, and I share my knowledge on CSEstack.org.
