Comment ajouter une ligne à une matrice dans NumPy (avec exemples)

Vous pouvez utiliser la syntaxe suivante pour ajouter une ligne à une matrice dans NumPy :

#add new_row to current_matrix
current_matrix = np.vstack([current_matrix, new_row])

Vous pouvez également utiliser la syntaxe suivante pour ajouter uniquement des lignes à une matrice qui remplissent une certaine condition :

#only add rows where first element is less than 10
current_matrix = np.vstack((current_matrix, new_rows[new_rows[:,0] < 10]))

Les exemples suivants montrent comment utiliser cette syntaxe dans la pratique.

Exemple 1 : ajouter une ligne à la matrice dans NumPy

Le code suivant montre comment ajouter une nouvelle ligne à une matrice dans NumPy :

import numpy as np

#define matrix
current_matrix = np.array([[1 ,2 ,3], [4, 5, 6], [7, 8, 9]])

#define row to add
new_row = np.array([10, 11, 12])

#add new row to matrix
current_matrix = np.vstack([current_matrix, new_row])

#view updated matrix
current_matrix

array([[ 1,  2,  3],
       [ 4,  5,  6],
       [ 7,  8,  9],
       [10, 11, 12]])

Notez que la dernière ligne a été ajoutée avec succès à la matrice.

Exemple 2 : ajouter des lignes à la matrice en fonction de la condition

Le code suivant montre comment ajouter plusieurs nouvelles lignes à une matrice existante en fonction d’une condition spécifique :

import numpy as np

#define matrix
current_matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

#define potential new rows to add
new_rows = np.array([[6, 8, 10], [8, 10, 12], [10, 12, 14]])

#only add rows where first element in row is less than 10
current_matrix = np.vstack((current_matrix, new_rows[new_rows[:,0] < 10]))

#view updated matrix
current_matrix

array([[ 1,  2,  3],
       [ 4,  5,  6],
       [ 7,  8,  9],
       [ 6,  8, 10],
       [ 8, 10, 12]])

Seules les lignes dont le premier élément était inférieur à 10 ont été ajoutées.

Remarque : Vous pouvez trouver la documentation complète en ligne de la fonction vstack() ici .

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres opérations courantes dans NumPy :

Comment trouver l’indice de valeur dans le tableau NumPy
Comment ajouter un tableau Numpy au Pandas DataFrame
Comment convertir un tableau NumPy en Pandas DataFrame