numpy.sum() in Python

Die Python NumPy sum() Funktion wird verwendet, um die Summe der Array-Elemente über eine gegebene Achse zu erhalten.

Python NumPy sum() Funktion Syntax

Die Syntax der Python NumPy sum() Methode lautet wie folgt:

sum(array, axis, dtype, out, keepdims, initial)

• Die Array-Elemente werden zur Berechnung der Summe verwendet.
• Wenn die Achse nicht angegeben ist, wird die Summe aller Elemente zurückgegeben. Wenn die Achse ein Tupel von Ganzzahlen ist, wird die Summe aller Elemente in den angegebenen Achsen zurückgegeben.
• Wir können dtype angeben, um den zurückgegebenen Ausgabedatentyp festzulegen.
• Die Variable out wird verwendet, um das Array zu spezifizieren, in dem das Ergebnis platziert werden soll. Es ist ein optionales Parameter.
• Der Parameter keepdims ist ein boolescher Parameter. Wenn dieser auf True gesetzt ist, werden die Achsen, die reduziert wurden, im Ergebnis als Dimensionen mit der Größe eins belassen.
• Der Parameter initial gibt den Startwert für die Summe an.

Beispiele für Python NumPy sum()

Werfen wir einen Blick auf einige Beispiele der numpy sum() Funktion.

1. Summe aller Elemente im Array

Wenn wir nur das Array in die Funktion sum() übergeben, wird es abgeflacht und die Summe aller Elemente wird zurückgegeben.

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4],
     [5, 6]])

total = np.sum(array1)
print(f'Sum of all the elements is {total}')

Ausgabe: Summe aller Elemente beträgt 21

2. Summe der Array-Elemente entlang der Achse

Wenn wir den Achsenwert angeben, wird die Summe der Elemente entlang dieser Achse zurückgegeben. Wenn die Form des Arrays (X, Y) ist, wird die Summe entlang der 0-Achse die Form (1, Y) haben. Die Summe entlang der 1-Achse wird die Form (1, X) haben.

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4],
     [5, 6]])

total_0_axis = np.sum(array1, axis=0)
print(f'Sum of elements at 0-axis is {total_0_axis}')

total_1_axis = np.sum(array1, axis=1)
print(f'Sum of elements at 1-axis is {total_1_axis}')

Ausgabe:

Sum of elements at 0-axis is [ 9 12]
Sum of elements at 1-axis is [ 3  7 11]

3. Festlegen des Datentyps für die Summe

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4]])

total_1_axis = np.sum(array1, axis=1, dtype=float)
print(f'Sum of elements at 1-axis is {total_1_axis}')

Ausgabe: Summe der Elemente entlang der 1-Achse ist [3. 7.]

4. Anfangswert für die Summe

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4]])

total_1_axis = np.sum(array1, axis=1, initial=10)
print(f'Sum of elements at 1-axis is {total_1_axis}')

Ausgabe: Summe der Elemente entlang der 1-Achse ist [13 17] Referenz: API-Dokumentation