Comment obtenir une colonne spécifique à partir d’un tableau NumPy (avec exemples)

Vous pouvez utiliser la syntaxe suivante pour obtenir une colonne spécifique d’un tableau NumPy :

#get column in index position 2 from NumPy array
my_array[:, 2]

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

Exemple 1 : obtenir une colonne du tableau NumPy

Le code suivant montre comment obtenir une colonne spécifique d’un tableau NumPy :

import numpy as np

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

#view NumPy array
data

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

#get column in index position 2
data[:, 2]

array([ 3,  7, 11])

Si vous souhaitez obtenir une colonne d’un tableau NumPy et la récupérer en tant que vecteur de colonne , vous pouvez utiliser la syntaxe suivante :

#get column in index position 2 (as a column vector)
data[:, [2]]

array([[ 3],
       [ 7],
       [11]])

Exemple 2 : obtenir plusieurs colonnes à partir du tableau NumPy

Le code suivant montre comment obtenir plusieurs colonnes d’un tableau NumPy :

import numpy as np

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

#view NumPy array
data

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

#get columns in index positions 1 and 3 from NumPy array
data[:, [1,3]]

array([[ 2,  4],
       [ 6,  8],
       [10, 12]])

Exemple 3 : obtenir les colonnes comprises dans la plage à partir du tableau NumPy

Le code suivant montre comment obtenir des colonnes dans une plage à partir d’un tableau NumPy :

import numpy as np

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

#view NumPy array
data

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

#get columns in index positions 0 through 3 (not including 3)
data[:, 0:3]

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

Notez que la dernière valeur de la plage (dans ce cas, 3) n’est pas incluse dans la plage de colonnes renvoyée.

Ressources additionnelles

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

Comment mapper une fonction sur un tableau NumPy
Comment ajouter une colonne à un tableau NumPy