How can I find the final cumulative sum across numpy axis? [duplicate]

Answer 1

You are almost there, but as you have it now, you are selecting just the final row. What you need is to select all rows from the last column, so your indexing at the end should be: [:,-1].

Example:

>>> a
array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19],
       [20, 21, 22, 23, 24]])

>>> a.cumsum(axis=1)[:,-1]
array([ 10,  35,  60,  85, 110])

Note, I'm leaving this up as I think it explains what was going wrong with your attempt, but admittedly, there are more effective ways of doing this in the other answers!

Answer 2

You can use numpy.ufunc.reduce if you don't need the intermediary accumulated results of any ufunc.

>>> a = np.arange(9).reshape(3,3)
>>> a
>>> 
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> 
>>> np.add.reduce(a, axis=1)
>>> array([ 3, 12, 21])

However, in the case of sum, Willem's answer is clearly superior and to be preferred. Just keep in mind that in the general case, there's ufunc.reduce.

Answer 3

The final cumulative sum of every row, is in fact simply the sum of every row, or the row-wise sum, so we can implement this as:

>>> x.sum(axis=1)
array([ 10,  35,  60,  85, 110])

So here for every row, we calculate the sum of all the columns. We thus do not need to first generate the sums in between (well these will likely be stored in an accumulator in numpy), but not "emitted" in the array.