numpy.cumsum() in Python

Die Python numpy cumsum() Funktion gibt die kumulative Summe der Elemente entlang der angegebenen Achse zurück.

Python numpy cumsum() Syntax

Die Syntax der cumsum()-Methode ist:

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

• Das Array kann ndarray oder array-ähnliche Objekte wie verschachtelte Listen sein.
• Der axis-Parameter definiert die Achse, entlang der die kumulative Summe berechnet wird. Wenn die Achse nicht angegeben ist, wird das Array abgeflacht und die kumulative Summe für das Ergebnisarray berechnet.
• Der dtype-Parameter definiert den Ausgabedatentyp, wie z.B. float und int.
• Der out optionale Parameter wird verwendet, um das Array für das Ergebnis anzugeben.

Python numpy cumsum() Beispiele

Lassen Sie uns einige Beispiele für die Berechnung der kumulativen Summe von Numpy-Array-Elementen ansehen.

1. Kumulative Summe der Numpy-Array-Elemente ohne Achse

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

Ausgabe: Kumulative Summe aller Elemente ist [ 1 3 6 10 15 21] Hier wird das Array zuerst auf [ 1 2 3 4 5 6] abgeflacht. Dann wird die kumulative Summe berechnet, was zu [ 1 3 6 10 15 21] führt.

2. Kumulative Summe entlang der Achse

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

Ausgabe :

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. Festlegen des Datentyps für das kumulative Summen-Array

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

Ausgabe :

Cumulative Sum of elements at 1-axis is:
[[ 1.  3.]
 [ 3.  7.]
 [ 5. 11.]]

Referenz: API-Dokument