objectif

Opérations vectorisées avec élargissement automatique des dimensions.

code minimal

import numpy as np

A = np.ones((3,1))
B = np.arange(3)
print((A + B).shape)

utilisation

import numpy as np

X = np.array([[1,2],[3,4]])
mu = X.mean(axis=0)
print((X - mu).tolist())

variante(s) utile(s)

import numpy as np

X = np.arange(6).reshape(2,3)
w = np.array([1,0,-1])
print((X * w).tolist())

notes

• Le broadcasting évite les boucles Python lentes.