Функция numpy.cumsum() в Python возвращает кумулятивную сумму элементов по заданной оси. Совокупная (кумулятивная) сумма — это сумма всех предыдущих элементов массива. Например, совокупная сумма массива [1, 2, 3] равна [1, 3, 6].
Синтаксис
|
1 |
numpy.cumsum(arr, axis=None, dtype=None, out=None) |
Параметры
Функция np.cumsum() имеет один обязательный аргумент и три необязательных аргумента в качестве параметров:
- arr: в этом аргументе передается массив. Это значение является аргументом, необходимым для возврата совокупной суммы массива. Массив задается внутри функции cumsum() для нахождения совокупной суммы этого массива.
- axis: ось, вдоль которой вычисляется кумулятивная сумма, является значением этого аргумента оси. Эта ось по умолчанию имеет значение «None». Однако ось можно изменить.
- dtype: dtype означает тип данных. Это определяет тип выходных данных. Выходные данные будут возвращены в этом указанном типе.
- out: это место назначения, где хранится выходной массив. Указанный размер массива должен соответствовать выходным данным, полученным от функции.
Возвращаемое значение
Функция np.cumsum() возвращает массив в качестве вывода. Функция cumsum() находит кумулятивную сумму для массива и возвращает ее в формате массива.
Пример 1 с numpy.cumsum()
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# Importing numpy as np import numpy as np # Creating an numpy array called arr arr = np.array([5, 6, 7, 8, 9, 10, 11, 12, 19, 25]) # Printing the array arr print("The array arr is: ") print(arr) # Printing the shape of the array print("The shape of the array arr is : ", arr.shape) # Finding the sum from the cumulative sum function for the array res = np.cumsum(arr)[-1] # Printing the sum of the array print("The sum of the array using the cumulative sum function is:") print(res) |
Выход
|
1 2 3 4 5 |
The array arr is: [ 5 6 7 8 9 10 11 12 19 25] The shape of the array arr is :(10,) The sum of the array using the cumulative sum function is: 112 |
Мы импортировали пакет numpy, который предоставляет несколько функций массива в этой программе. Затем мы создали массив numpy с именем arr, используя функцию с именем np.array(). Этот массив arr состоит из нескольких целочисленных элементов. Для этого массива сумма вычисляется с помощью функции кумулятивной суммы.
Для этого мы передали массив в функцию cumsum(). Функция cumsum() находит кумулятивную сумму, после чего мы возвращаем только последнее значение из массива по индексу -1. Таким образом, он возвращает последний элемент массива.
Последний элемент массива представляет собой сумму всех элементов. Следовательно, мы напечатали сумму всех элементов, используя функцию cumsum().
Пример 2
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
# Importing numpy as np import numpy as np # Creating an numpy array called a a = np.array([5, 6, 7, 8, 9, 10, 11, 12, 19, 25]) # printing the array a print("The array a is: ") print(a) # Printing the shape of the array a print("The shape of the array arr is : ", a.shape) # Finding the cumulative sum of the single dimensional array op = np.cumsum(a) # Printing the sum of the array print("The cumulative sum of the array is: ") print(op) # Creating an numpy array called arr arr = np.array([[1, 4, 0], [7, 0, 9], [5, 0, 3], [12, 0, 5]]) # printing the array arr print("The array arr is: ") print(arr) # Printing the shape of the array print("The shape of the array arr is : ", arr.shape) # Finding the cumulative sum for the two dimensional array res = np.cumsum(arr, axis=1) # Printing the cumulative sum of the array print("The cumulative sum of the array is: ") print(res) |
Выход
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
The array a is: [ 5 6 7 8 9 10 11 12 19 25] The shape of the array arr is :(10,) The cumulative sum of the array is: [ 5 11 18 26 35 45 56 68 87 112] The array arr is: [[ 1 4 0] [ 7 0 9] [ 5 0 3] [12 0 5]] The shape of the array arr is :(4, 3) The cumulative sum of the array is: [[ 1 5 5] [ 7 7 16] [ 5 5 8] [12 12 17]] |
Мы создали два пустых массива с именами a и arr, используя функцию np.array(). Массив «а» — это одномерный массив. Этот массив состоит из некоторых целочисленных значений. Мы передали этот массив в функцию np.cumsum(). Функция np.cumsum() вычисляет совокупную сумму массива. Результат отображается.
Массив arr состоит из четырех строк, по 3 столбца в каждой. Затем мы передали этот массив функции np.cumsum() с осью, равной 1. Функция cumsum() вычисляет совокупную сумму и возвращает массив.
