// Tutorial //

numpy.cumsum() in Python

Published on August 3, 2022
Default avatar

By Pankaj

numpy.cumsum() in Python

While we believe that this content benefits our community, we have not yet thoroughly reviewed it. If you have any suggestions for improvements, please let us know by clicking the “report an issue“ button at the bottom of the tutorial.

Python numpy cumsum() function returns the cumulative sum of the elements along the given axis.

Python numpy cumsum() syntax

The cumsum() method syntax is:

cumsum(array, axis=None, dtype=None, out=None)
  • The array can be ndarray or array-like objects such as nested lists.
  • The axis parameter defines the axis along which the cumulative sum is calculated. If the axis is not provided then the array is flattened and the cumulative sum is calculated for the result array.
  • The dtype parameter defines the output data type, such as float and int.
  • The out optional parameter is used to specify the array for the result.

Python numpy cumsum() Examples

Let’s look at some examples of calculating cumulative sum of numpy array elements.

1. Cumulative Sum of Numpy Array Elements without axis

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4],
     [5, 6]])

total = np.cumsum(array1)
print(f'Cumulative Sum of all the elements is {total}')

Output: Cumulative Sum of all the elements is [ 1 3 6 10 15 21] Here, the array is first flattened to [ 1 2 3 4 5 6]. Then the cumulative sum is calculated, resulting in [ 1 3 6 10 15 21].

2. Cumulative Sum along the axis

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4],
     [5, 6]])

total_0_axis = np.cumsum(array1, axis=0)
print(f'Cumulative Sum of elements at 0-axis is:\n{total_0_axis}')

total_1_axis = np.cumsum(array1, axis=1)
print(f'Cumulative Sum of elements at 1-axis is:\n{total_1_axis}')

Output:

Cumulative Sum of elements at 0-axis is:
[[ 1  2]
 [ 4  6]
 [ 9 12]]
Cumulative Sum of elements at 1-axis is:
[[ 1  3]
 [ 3  7]
 [ 5 11]]

3. Specifying data type for the cumulative sum array

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4],
     [5, 6]])

total_1_axis = np.cumsum(array1, axis=1, dtype=float)
print(f'Cumulative Sum of elements at 1-axis is:\n{total_1_axis}')

Output:

Cumulative Sum of elements at 1-axis is:
[[ 1.  3.]
 [ 3.  7.]
 [ 5. 11.]]

Reference: API Doc

Thanks for learning with the DigitalOcean Community. Check out our offerings for compute, storage, networking, and managed databases.

Learn more about us


About the authors
Default avatar
Pankaj

author

Still looking for an answer?

Ask a questionSearch for more help

Was this helpful?
 

Try DigitalOcean for free

Click below to sign up and get $200 of credit to try our products over 60 days!

Sign up

card icon
Get our biweekly newsletter

Sign up for Infrastructure as a Newsletter.

Sign up
card icon
Hollie's Hub for Good

Working on improving health and education, reducing inequality, and spurring economic growth? We’d like to help.

Learn more
card icon
Become a contributor

You get paid; we donate to tech nonprofits.

Learn more
Welcome to the developer cloud

DigitalOcean makes it simple to launch in the cloud and scale up as you grow – whether you’re running one virtual machine or ten thousand.

Learn more ->
DigitalOcean Cloud Control Panel
Get started for free

Enter your email to get $200 in credit for your first 60 days with DigitalOcean.

New accounts only. By submitting your email you agree to our Privacy Policy.

© 2023 DigitalOcean, LLC.