Exemples de comment calculer la somme le long d'un axe donné avec numpy en python:
Créer une matrice aléatoire avec numpy
Créons une matrice aléatoire avec numpy de taille = (3,3):
import numpy as np
data = np.random.randint(0,5,size=(3,3))
donne par exemple
array([[4, 1, 2],
[2, 1, 1],
[4, 0, 1]])
Sommer le long du premier axe (axis=0)
Pour sommer le long d'un axe donné d'une matrice avec numpy, une solution consiste à utiliser numpy.sum avec le paramètre "axis":
data.sum(axis=0)
donne
array([10, 2, 4])
Sommer le long du second axe (axis=1)
Un autre exemple le long de axis=1
data.sum(axis=1)
donne
array([7, 4, 5])
Additionner tous les éléments d'une matrice avec numpy
Note: pour additionner tous les éléments d'une matrice avec numpy :
data.sum()
donne ici
16
Un autre exemple avec plus de 2 dimensions
Un autre exemple avec une matrice de taille = (3,3,3)
import numpy as np
data = np.random.randint(0,5,size=(3,3,3))
donne par exemple
array([[[4, 2, 4],
[3, 3, 1],
[0, 4, 1]],
[[3, 4, 0],
[3, 1, 3],
[4, 4, 2]],
[[2, 0, 2],
[2, 4, 1],
[3, 3, 4]]])
Sommer le long axis=0
data.sum(axis=0)
donne
array([[ 9, 6, 6],
[ 8, 8, 5],
[ 7, 11, 7]])
Sommer le long axis=1
data.sum(axis=1)
donne
array([[ 7, 9, 6],
[10, 9, 5],
[ 7, 7, 7]])
Sommer le long axis=2
data.sum(axis=2)
donne
array([[10, 7, 5],
[ 7, 7, 10],
[ 4, 7, 10]])