NumPy : Comment obtenir des indices dont la valeur est vraie

Vous pouvez utiliser les méthodes suivantes pour obtenir les indices pour lesquels une condition est vraie dans NumPy :

Méthode 1 : obtenir les indices où la condition est vraie dans le tableau NumPy

#get indices of values greater than 10
np.asarray(my_array>10).nonzero()

Méthode 2 : obtenir les indices où la condition est vraie dans la matrice NumPy

#get indices of values greater than 10
np.transpose((my_matrix>10).nonzero())

Méthode 3 : obtenir des indices où la condition est vraie dans n’importe quelle ligne de la matrice NumPy

#get indices of rows where any value is greater than 10
np.asarray(np.any(my_matrix>10, axis=1)).nonzero()

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

Exemple 1 : obtenir des indices où la condition est vraie dans le tableau NumPy

Le code suivant montre comment obtenir tous les indices d’un tableau NumPy dont la valeur est supérieure à 10 :

import numpy as np

#create NumPy array
my_array = np.array([2, 2, 4, 5, 7, 9, 11, 12, 3, 19])

#get index of values greater than 10
np.asarray(my_array>10).nonzero()

(array([6, 7, 9], dtype=int32),)

À partir du résultat, nous pouvons voir que les valeurs aux positions d’index 6 , 7 et 9 du tableau NumPy d’origine ont des valeurs supérieures à 10.

Exemple 2 : obtenir des indices où la condition est vraie dans la matrice NumPy

Le code suivant montre comment obtenir tous les indices d’une matrice NumPy dont la valeur est supérieure à 10 :

import numpy as np

#create NumPy matrix
my_matrix = np.array([[2, 5, 9, 12],
                     [6, 7, 8, 8],
                     [2, 5, 7, 8],
                     [4, 1, 15, 11]])

#get index of values greater than 10
np.transpose((my_matrix>10).nonzero())

array([[0, 3],
       [3, 2],
       [3, 3]], dtype=int32)

À partir du résultat, nous pouvons voir que les valeurs dans les positions d’index suivantes de la matrice ont des valeurs supérieures à 10 :

• [0, 3]
• [3, 2]
• [3, 3]

Exemple 3 : obtenir des indices où la condition est vraie dans n’importe quelle ligne de la matrice NumPy

Le code suivant montre comment obtenir tous les indices de ligne dans une matrice NumPy où la valeur est supérieure à 10 dans n’importe quel élément de la ligne :

import numpy as np

#create NumPy matrix
my_matrix = np.array([[2, 5, 9, 12],
                     [6, 7, 8, 8],
                     [2, 5, 7, 8],
                     [4, 1, 15, 11]])

#get index of rows where any value is greater than 10
np.asarray(np.any(my_matrix>10, axis=1)).nonzero()

(array([0, 3], dtype=int32),)

À partir du résultat, nous pouvons voir que les lignes 0 et 3 ont au moins une valeur supérieure à 10.

Remarque : Pour obtenir des indices où une condition est vraie dans une colonne, utilisez plutôt axis=0 .

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres tâches courantes dans NumPy :

Comment remplir un tableau NumPy avec des valeurs
Comment supprimer des éléments spécifiques du tableau NumPy
Comment remplacer des éléments dans un tableau NumPy
Comment obtenir une ligne spécifique à partir d’un tableau NumPy