Trouver le premier index d'élément dans NumPy Array

Manav Narula 30 janvier 2023
  1. Utilisez la fonction where() pour trouver le premier index d’un élément dans un tableau NumPy
  2. Utilisez la fonction nonzero() pour trouver le premier index d’un élément dans un tableau NumPy
  3. Utilisez la fonction argmax() pour trouver le premier index d’un élément dans un tableau NumPy
  4. Utilisez la fonction index() pour trouver le premier index d’un élément dans un tableau NumPy
Trouver le premier index d'élément dans NumPy Array

Dans ce didacticiel, nous expliquerons comment trouver le premier index d’un élément dans un tableau numpy.

Utilisez la fonction where() pour trouver le premier index d’un élément dans un tableau NumPy

La fonction where() du module numpy permet de renvoyer un tableau contenant les indices des éléments qui satisfont certaines conditions. La condition est spécifiée dans la fonction.

Nous pouvons l’utiliser pour trouver le premier index d’une valeur spécifique dans un tableau, comme indiqué ci-dessous.

a = np.array([7, 8, 9, 5, 2, 1, 5, 6, 1])

print(np.where(a == 1)[0][0])

Production:

5

Utilisez la fonction nonzero() pour trouver le premier index d’un élément dans un tableau NumPy

La fonction nonzero() renvoie les indices de tous les éléments non nuls dans un tableau numpy. Il renvoie des tuples de plusieurs tableaux pour un tableau multidimensionnel.

Similaire à la fonction where(), nous pouvons également spécifier la condition afin qu’elle puisse également renvoyer la position d’un élément spécifique.

Par example,

a = np.array([7, 8, 9, 5, 2, 1, 5, 6, 1])

print(np.nonzero(a == 1)[0][0])

Production:

5

Pour les besoins les plus élémentaires, les fonctions where() et nonzero() semblent similaires. La différence survient avec la fonction where() lorsque vous souhaitez choisir des éléments du tableau a si une condition est True et du tableau b lorsque cette condition est False.

Utilisez la fonction argmax() pour trouver le premier index d’un élément dans un tableau NumPy

argmax() trouve l’index de l’élément maximum dans un tableau. Nous pouvons spécifier la condition d’égalité dans la fonction et trouver également l’index de l’élément requis.

Par example,

a = np.array([7, 8, 9, 5, 2, 1, 5, 6, 1])

print(np.argmax(a == 1))

Production:

5

Utilisez la fonction index() pour trouver le premier index d’un élément dans un tableau NumPy

Dans cette méthode, nous allons d’abord convertir le tableau en liste en utilisant la fonction tolist(). Ensuite, nous utiliserons la fonction index(), qui retournera la position de l’élément spécifié.

Le code suivant implémente cela.

a = np.array([7, 8, 9, 5, 2, 1, 5, 6, 1])

print(a.tolist().index(1))

Production:

5
Auteur: Manav Narula
Manav Narula avatar Manav Narula avatar

Manav is a IT Professional who has a lot of experience as a core developer in many live projects. He is an avid learner who enjoys learning new things and sharing his findings whenever possible.

LinkedIn