numpy.cumsum () в Python

Функция Python Numpy Cumsum () возвращает совокупную сумму элементов вдоль заданной оси.

Python Numpy Cumsum () синтаксис

Синтаксис метода CumSum ():

cumsum(array, axis=None, dtype=None, out=None)

массив Может быть NDARRay или Array-подобные объекты, такие как вложенные списки.
Ось Параметр определяет ось, вдоль которой рассчитана совокупная сумма. Если ось не предоставлена, то массив сплющен, и совокупная сумма рассчитывается для массива результатов.
dtype Параметр определяет тип выходных данных, таких как float и int.
OUT Дополнительный параметр используется для указания массива для результата.

Python Numpy Cumsum () Примеры

Давайте посмотрим на некоторые примеры расчета совокупной суммы элементов Numpy Array.

1. Совокупная сумма гампиальных элементов массива без оси

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}')

Выход : Совокупная сумма всех элементов [1 3 6 10 15 21]

Здесь массив сначала сплющен до [1 2 3 4 5 6]. Затем совокупная сумма рассчитывается, что приводит к [1 3 6 10 15 21].

2. Совокупная сумма вдоль оси

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}')

Выход:

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. Указание типа данных для кумулятивного массива суммирования

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}')