Website-Suche

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 Methode sum() von Python NumPy lautet:

sum(array, axis, dtype, out, keepdims, initial)
  • Die Array-Elemente werden verwendet, um die Summe zu berechnen.
  • 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 Datenausgabetyp zu spezifizieren.
  • Die out-Variable wird verwendet, um das Array anzugeben, in das das Ergebnis platziert werden soll. Es ist ein optionaler Parameter.
  • Der keepdims ist ein boolescher Parameter. Wenn dieser auf True gesetzt ist, bleiben die Achsen, die reduziert werden, im Ergebnis als Dimensionen mit der Größe eins erhalten.
  • Der anfängliche Parameter gibt den Startwert für die Summe an.

Python numpy sum() Beispiele

Lassen Sie uns einige Beispiele für die numpy sum() Funktion ansehen.

1. Summe aller Elemente im Array

Wenn wir nur das Array in die sum()-Funktion ü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}')

Die 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 Array-Form (X, Y) ist, hat die Summe entlang der 0-Achse die Form (1, Y). Die Summe entlang der 1-Achse hat die Form (1, X).

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. Ausgabe-Datentyp der Summe festlegen

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: Die Summe der Elemente an der 1-Achse ist [13 17] Referenz: API-Dokumentation