Ajout de colonnes supplémentaires à un tableau NumPy

Supposons que vous ayez un tableau NumPy 2D a comme suit :

a = np.array([
    [1, 2, 3],
    [2, 3, 4],
])

Pour ajouter une colonne de zéros le long du deuxième axe, vous pouvez utiliser différentes méthodes. Une approche consiste à utiliser la fonction np.c_[ ] :

b = np.c_[a, np.zeros(a.shape[0])]

Cela créera un nouveau tableau b avec une colonne supplémentaire de zéros :

b = np.array([
    [1, 2, 3, 0],
    [2, 3, 4, 0],
])

Vous pouvez également utiliser la fonction np.r_[ ] :

b = np.r_[a, np.zeros((a.shape[0], 1))]

Cette méthode ajoutera également une colonne de zéros au tableau.

Notez que np.r_[ ] et np.c_[ ] fournit des options flexibles pour modifier les dimensions du tableau. Ils peuvent être utilisés pour mélanger des vecteurs et des scalaires, ajouter des lignes ou des colonnes et même insérer des tableaux entiers à des positions spécifiées.