Recherche des tableaux NumPy

Recherche de tableau

Vous pouvez rechercher (rechercher) une valeur dans le tableau et retourner les indices correspondants obtenus.

Pour rechercher un tableau, utilisez where() méthode.

Exemple

Recherche des indices des valeurs 4 :

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 4, 4])
x = np.where(arr == 4)
print(x)

Exécuter un exemple

Dans cet exemple, il retournera un tuple :(array([3, 5, 6],)

Cela signifie que la valeur 4 apparaît aux indices 3, 5 et 6.

Exemple

Recherche des indices des valeurs paires :

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8])
x = np.where(arr%2 == 0)
print(x)

Exécuter un exemple

Exemple

Recherche des indices des valeurs impaires :

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8])
x = np.where(arr%2 == 1)
print(x)

Exécuter un exemple

recherche de tri

Il y a un nom appelé searchsorted() une méthode, cette méthode effectue une recherche binaire dans le tableau et retourne l'index où l'on peut insérer la valeur spécifiée pour maintenir l'ordre de recherche.

présumé searchsorted() Cette méthode est utilisée pour trier les tableaux.

Exemple

Trouver l'index où insérer la valeur 7 :

import numpy as np
arr = np.array([6, 7, 8, 9])
x = np.searchsorted(arr, 7)
print(x)

Exécuter un exemple

Explication de l'exemple :Le nombre 7 doit être inséré à l'index 1 pour maintenir l'ordre.

Cette méthode cherche à partir de la gauche et retourne le premier index où le nombre 7 n'est plus supérieur au prochain nombre.

Recherche de droite

Par défaut, le plus petit index est retourné, mais nous pouvons donner side='right'afin de retourner l'index le plus à droite.

Exemple

Chercher l'index où insérer la valeur 7 en commençant par la droite :

import numpy as np
arr = np.array([6, 7, 8, 9])
x = np.searchsorted(arr, 7, side='right')
print(x)

Exécuter un exemple

Explication de l'exemple :Le nombre 7 doit être inséré à l'index 2 pour maintenir l'ordre.

Cette méthode cherche à partir de la droite et retourne le premier index où le nombre 7 n'est plus inférieur au prochain nombre.

Valeurs multiples

Pour chercher plusieurs valeurs, utilisez un tableau avec des valeurs spécifiées.

Exemple

Trouver les indices où insérer les valeurs 2, 4 et 6 :

import numpy as np
arr = np.array([1, 3, 5, 7])
x = np.searchsorted(arr, [2, 4, 6])
print(x)

Exécuter un exemple

La valeur de retour est un tableau :[1 2 3] Inclut trois indices, où 2, 4 et 6 seront insérés dans l'array original pour maintenir l'ordre.