Comment ajouter une colonne à un tableau NumPy ?

Comment augmenter les tableaux NumPy avec une colonne supplémentaire

Lorsque vous travaillez avec des tableaux NumPy, il est souvent nécessaire d'ajouter une colonne supplémentaire à un tableau existant. Ceci peut être réalisé en utilisant différentes méthodes, dont l'une consiste à utiliser la fonction np.c_[...] de NumPy.

Considérez le tableau 2D suivant :

<code class="python">a = np.array([
    [1, 2, 3],
    [2, 3, 4],
])</code>

Pour ajouter une colonne de zéros à ce tableau, nous pouvons utiliser np.c_[...] :

<code class="python">b = np.c_[a, np.zeros(a.shape[0])]</code>

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

<code class="python">b = np.array([
    [1, 2, 3, 0],
    [2, 3, 4, 0],
])</code>

Une autre alternative pour ajouter des colonnes consiste à utiliser la fonction np.r_[...]. Il peut être utilisé pour ajouter des lignes et des colonnes à des tableaux, offrant ainsi une plus grande flexibilité.

Par exemple, pour ajouter une colonne supplémentaire :

<code class="python">c = np.c_[a, np.ones(a.shape[0])]</code>

Pour ajouter plusieurs colonnes :

<code class="python">d = np.c_[a, 2*np.ones(a.shape[0]), 3*np.ones(a.shape[0])]</code>

La notation entre crochets [...] dans ces fonctions permet des ajouts de colonnes flexibles en spécifiant des valeurs ou des tableaux aux emplacements souhaités.

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!