Arrays, formes et axes

Forme = contrat

Un array numpy a une forme (n, p) : n lignes, p colonnes. X[0] = première ligne, X[:, 1] = deuxième colonne, X.mean(axis=0) = moyenne par colonne.

import numpy as np
X = np.array([[1., 2.], [3., 4.], [5., 6.]])
print(X.shape)          # (3, 2)
print(X[:, 0].mean())   # 3.0 = moyenne colonne 0
print(X.mean(axis=0))   # [3. 4.]

Erreur classique

Mélanger (3,) et (3, 1) : le premier est un vecteur, le second une colonne. X.mean(axis=1) donne une valeur par ligne. Vérifiez toujours shape avant @ : (3, 2) @ (2,) donne (3,).