Trouver le premier index d'élément dans NumPy Array
-
Utilisez la fonction
where()
pour trouver le premier index d’un élément dans un tableau NumPy -
Utilisez la fonction
nonzero()
pour trouver le premier index d’un élément dans un tableau NumPy -
Utilisez la fonction
argmax()
pour trouver le premier index d’un élément dans un tableau NumPy -
Utilisez la fonction
index()
pour trouver le premier index d’un élément dans un tableau NumPy
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
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