numpy.sum() en Python

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

Syntaxe de la fonction sum() de Python numpy

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 renvoyée. Si l'axe est un tuple d'entiers, la somme de tous les éléments dans les axes donnés est renvoyée.
• Nous pouvons spécifier dtype pour définir le type de données de sortie retourné.
• La variable out est utilisée pour spécifier le tableau dans lequel placer le résultat. C'est un paramètre optionnel.
• Le keepdims est un paramètre booléen. S'il est défini sur True, les axes qui sont réduits sont conservé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 Python numpy

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

La 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 renvoyée. Si la forme du tableau est (X, Y), alors la somme le long de l'axe 0 aura pour forme (1, Y). La somme le long de l'axe 1 aura pour 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écification du 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 : La 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 à l'axe 1 est [13 17] Référence: Documentation API