Sequentially add elements in numpy array into new array

Question

I'm curious if there is a built in function to transform an array of values into a cumulative array of values.

Example:

input = np.asarray([0.000,1.500,2.100,5.000])

into

[0.000,1.500,3.600,8.600]

Thanks!

Sheldore · Accepted Answer · 2019-01-27 17:52:48Z

1

Use in-built cumsum from NumPy to get the cumulative sum of your array inputt as

inputt = np.asarray([0.000,1.500,2.100,5.000])
print (np.cumsum(inputt)) 

# [0.  1.5 3.6 8.6]

I renamed your array because input is already an in-built function in python to get the user input from the keyboard

answered Jan 27, 2019 at 17:52

Sheldore

39.2k9 gold badges63 silver badges76 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

out of curiosity do you know how to truncate it at 100?

@jparanich: 8.6 and 8.6000 are equivalent. But still you can control the number of digits after decimal while printing them. See here

Sorry I should have been clearer. I meant truncate the cumulative rolling numbers at 100 so the sum does not exceed 100.