← retour aux snippets

numpy: règles de broadcasting

Appliquer des opérations sur formes compatibles sans copie.

python numpy #numpy#broadcast#shape

objectif

Appliquer des opérations sur formes compatibles sans copie.

code minimal

import numpy as np
a = np.ones((3,1)); b = np.arange(4)
c = a + b
print(c.shape == (3,4))

utilisation

import numpy as np
x = np.arange(6).reshape(3,2)
w = np.array([10,100])
print((x * w).tolist())

variante(s) utile(s)

import numpy as np
x = np.arange(3); y = 2
print((x + y).tolist())

notes

  • Les dimensions de taille 1 s’étendent automatiquement.