Comment ajouter des colonnes supplémentaires à un tableau NumPy ?

Ajout de colonnes supplémentaires à un tableau NumPy

Supposons que vous ayez un tableau NumPy 2D 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],
])

Alternativement, vous pouvez 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_[ ] fournissent 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.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!