Skip to content
Related Articles

Related Articles

Prefix sum array in Python using accumulate function

View Discussion
Improve Article
Save Article
  • Difficulty Level : Hard
  • Last Updated : 28 Jul, 2022
View Discussion
Improve Article
Save Article

We are given an array, find prefix sums of given array. Examples:

Input : arr = [1, 2, 3]
Output : sum = [1, 3, 6]

Input : arr = [4, 6, 12]
Output : sum = [4, 10, 22]

A prefix sum is a sequence of partial sums of a given sequence. For example, the cumulative sums of the sequence {a, b, c, …} are a, a+b, a+b+c and so on. We can solve this problem in python quickly using accumulate(iterable) method. 

Implementation:

Python3




# function to find cumulative sum of array
from itertools import accumulate
 
def cumulativeSum(input):
    print ("Sum :", list(accumulate(input)))
 
# Driver program
if __name__ == "__main__":
    input = [4, 6, 12]
    cumulativeSum(input)

Output

Sum : [4, 10, 22]
My Personal Notes arrow_drop_up
Recommended Articles
Page :

Start Your Coding Journey Now!