numpy.sum() en Python

La fonction sum() de numpy en Python est utilisée pour obtenir la somme des éléments d’un tableau sur un axe donné.

Syntaxe de la fonction sum() de numpy en Python

La syntaxe de la méthode sum() de Python NumPy est :

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

• Les éléments du tableau sont utilisés pour calculer la somme.
• Si l’axe n’est pas fourni, la somme de tous les éléments est retournée. Si l’axe est un tuple d’entiers, la somme de tous les éléments sur les axes donnés est retournée.
• Nous pouvons spécifier dtype pour spécifier le type de données de sortie retourné.
• La variable out est utilisée pour spécifier le tableau où placer le résultat. C’est un paramètre facultatif.
• Le paramètre keepdims est un paramètre booléen. S’il est défini sur True, les axes qui sont réduits sont laissés dans le résultat en tant que dimensions de taille un.
• Le paramètre initial spécifie la valeur de départ pour la somme.

Exemples de la fonction sum() de numpy en Python

Examinons quelques exemples de la fonction sum() de numpy.

1. Somme de tous les éléments dans le tableau

Si nous ne passons que le tableau dans la fonction sum(), il est aplati et la somme de tous les éléments est retournée.

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

Sortie: Somme de tous les éléments est 21

2. Somme des éléments du tableau le long de l’axe

Si nous spécifions la valeur de l’axe, la somme des éléments le long de cet axe est retournée. Si la forme du tableau est (X, Y), alors la somme le long de l’axe 0 sera de forme (1, Y). La somme le long de l’axe 1 sera de forme (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}')

Sortie:

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

3. Spécifier le type de données de sortie de la somme

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

Sortie: Somme des éléments à l'axe 1 est [3. 7.]

4. Valeur initiale pour la somme

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

Sortie: La somme des éléments sur l'axe 1 est [13 17] Référence: Documentation de l’API