numpy element sum privous code example

Example: numpy element sum privous

>>> a
array([[1, 2, 3],
       [4, 5, 6]])
       
>>> np.cumsum(a,axis=0)      # sum over rows for each of the 3 columns
array([[1, 2, 3],
       [5, 7, 9]])
>>> np.cumsum(a,axis=1)      # sum over columns for each of the 2 rows
array([[ 1,  3,  6],
       [ 4,  9, 15]])