Comment calculer la somme le long d'un axe donné avec numpy en python ?

Published: 21 septembre 2021

Tags: Python; Numpy;

DMCA.com Protection Status

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]])

Références