Comment découper un tableau NumPy 2D (avec exemples)

Vous pouvez utiliser les méthodes suivantes pour découper un tableau NumPy 2D :

Méthode 1 : sélectionner des lignes spécifiques dans le tableau NumPy 2D

#select rows in index positions 2 through 5
arr[2:5, :]

Méthode 2 : sélectionner des colonnes spécifiques dans le tableau NumPy 2D

#select columns in index positions 1 through 3
arr[:, 1:3]

Méthode 3 : sélectionner des lignes et des colonnes spécifiques dans le tableau NumPy 2D

#select rows in range 2:5 and columns in range 1:3
arr[2:5, 1:3]

Les exemples suivants montrent comment utiliser chaque méthode en pratique avec le tableau NumPy 2D suivant :

import numpy as np

#create NumPy array
arr = np.arange(24).reshape(6,4)

#view NumPy array
print(arr)

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]
 [12 13 14 15]
 [16 17 18 19]
 [20 21 22 23]]

Exemple 1 : sélectionner des lignes spécifiques d’un tableau NumPy 2D

Nous pouvons utiliser la syntaxe suivante pour sélectionner les lignes dans les positions d’index 2 à 5 :

#select rows in index positions 2 through 5
arr[2:5, :]

array([[ 8,  9, 10, 11],
       [12, 13, 14, 15],
       [16, 17, 18, 19]])

Notez que la syntaxe 2:5 indique à NumPy de sélectionner les lignes 2 à 5, mais n’inclut pas 5.

Ainsi, cette syntaxe sélectionne toutes les valeurs des lignes avec des positions d’index de 2, 3 et 4.

Exemple 2 : sélectionner des colonnes spécifiques du tableau NumPy 2D

Nous pouvons utiliser la syntaxe suivante pour sélectionner les colonnes aux positions d’index 1 à 3 :

#select columns in index positions 1 through 3
arr[, 1:3]

array([[ 1,  2],
       [ 5,  6],
       [ 9, 10],
       [13, 14],
       [17, 18],
       [21, 22]]))

Notez que la syntaxe 1:3 indique à NumPy de sélectionner les colonnes 1 à 3, mais n’inclut pas 3.

Ainsi, cette syntaxe sélectionne toutes les valeurs des colonnes avec des positions d’index de 1 et 2.

Exemple 3 : sélectionner des lignes et des colonnes spécifiques du tableau NumPy 2D

Nous pouvons utiliser la syntaxe suivante pour sélectionner les lignes dans les positions d’index 2 à 5 et les colonnes dans les positions d’index 1 à 3 :

#select rows in 2:5 and columns in 1:3
arr[2:5, 1:3]

array([[ 9, 10],
       [13, 14],
       [17, 18]])

Cette syntaxe renvoie toutes les valeurs du tableau NumPy 2D entre les positions d’index de ligne 2 à 5 et les positions d’index de colonne 1 à 3.

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 obtenir une colonne spécifique à partir d’un tableau NumPy
Comment ajouter une colonne à un tableau NumPy