Comment ajouter une colonne à un tableau NumPy (avec exemples)

Vous pouvez utiliser l’une des méthodes suivantes pour ajouter une colonne à un tableau NumPy :

Méthode 1 : ajouter une colonne à la fin du tableau

np.append(my_array, [[value1], [value2], [value3], ...], axis=1)

Méthode 2 : insérer une colonne dans une position spécifique du tableau

np.insert(my_array, 3, [value1, value2, value3, ...], axis=1) 

Les exemples suivants montrent comment utiliser chaque méthode dans la pratique.

Exemple 1 : ajouter une colonne à la fin du tableau NumPy

Supposons que nous ayons le tableau NumPy suivant :

import numpy as np

#create NumPy array
my_array = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])

#view NumPy array
my_array

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

Nous pouvons utiliser la syntaxe suivante pour ajouter une colonne à la fin du tableau NumPy :

#append column to end of NumPy array
new_array = np.append(my_array, [[10], [13]], axis=1)

#view updated array
new_array

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

Exemple 2 : Insérer une colonne dans une position spécifique du tableau NumPy

Supposons que nous ayons le tableau NumPy suivant :

import numpy as np

#create NumPy array
my_array = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

#view NumPy array
my_array

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

Nous pouvons utiliser la syntaxe suivante pour insérer une nouvelle colonne avant la colonne en position d’index 2 du tableau NumPy :

#insert new column before column in index position 2
new_array = np.insert(my_array, 2, [10, 13, 19], axis=1)

#view updated array
new_array

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

Notez que la nouvelle colonne de valeurs a été insérée avant la colonne en position d’index 2.

Ressources additionnelles

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

Comment ajouter un tableau Numpy à un DataFrame Pandas
Comment convertir Pandas DataFrame en tableau NumPy