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

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

#get row 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 ligne du tableau NumPy

Le code suivant montre comment obtenir une ligne 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
print(data)

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

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

array([ 9, 10, 11, 12])

Notez que seule la ligne en position d’index 2 du tableau NumPy est renvoyée.

Exemple 2 : obtenir plusieurs lignes à partir du tableau NumPy

Le code suivant montre comment obtenir plusieurs lignes 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 rows in index positions 0 and 2 from NumPy array
data[[0,2], :]

array([[ 1,  2,  3,  4],
       [ 9, 10, 11, 12]])

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

Le code suivant montre comment obtenir des lignes 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 rows in index positions 0 through 1
data[0:2, :]

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

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

Ressources additionnelles

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

Comment obtenir une colonne spécifique à partir d’un tableau NumPy
Comment mapper une fonction sur un tableau NumPy
Comment ajouter une colonne à un tableau NumPy