Функция cumsum в Python: примеры и объяснения

Функция cumsum в Python используется для вычисления кумулятивной суммы элементов последовательности. Кумулятивная сумма - это последовательность, в которой каждый элемент равен сумме всех предыдущих элементов, включая сам элемент.

Для использования функции cumsum можно использовать встроенную функцию sum и функцию accumulate из модуля itertools. Рассмотрим примеры использования обоих способов.

1. Использование функции sum:

python
numbers = [1, 2, 3, 4, 5]
cumulative_sum = []
current_sum = 0
for number in numbers:
    current_sum += number
    cumulative_sum.append(current_sum)
print(cumulative_sum)  # Вывод: [1, 3, 6, 10, 15]

2. Использование функции accumulate:

python
from itertools import accumulate
numbers = [1, 2, 3, 4, 5]
cumulative_sum = list(accumulate(numbers))
print(cumulative_sum)  # Вывод: [1, 3, 6, 10, 15]

Оба способа дают одинаковый результат - список с кумулятивной суммой элементов исходной последовательности. В первом примере мы проходим по исходному списку, вычисляем текущую сумму элементов и добавляем ее в новый список cumulative_sum. Во втором примере мы используем функцию accumulate, которая сама вычисляет кумулятивную сумму элементов последовательности и возвращает ее в виде итератора, который мы преобразуем в список с помощью функции list.

Таким образом, функция cumsum в Python позволяет вычислить кумулятивную сумму элементов последовательности с использованием функции sum или функции accumulate из модуля itertools.