Comment trouver le premier index d'un élément dans un tableau NumPy ?

Trouver le premier index d'un élément de tableau

Semblable aux listes Python, les tableaux NumPy offrent une méthode pour localiser la première occurrence d'un élément spécifique valeur. Cette question explore l'équivalent NumPy de la méthode list.index().

Méthode de récupération d'index de NumPy

Pour récupérer l'index du premier élément d'un tableau NumPy qui correspond à une valeur donnée, utilisez la fonction np.where :

itemindex = np.where(array == item)

Cette fonction renvoie un tuple contenant deux tableaux :

• Le premier tableau contient les indices de ligne des éléments correspondants.
• Le deuxième tableau contient les indices de colonne des éléments correspondants.

Ce tuple fournit les coordonnées nécessaires pour accéder à l'élément correspondant dans le tableau.

Exemple Utilisation

Considérons un tableau à deux dimensions :

array = np.array([[1, 2, 3], [4, 5, 6]])

Si on veut trouver la première occurrence de la valeur 5, on peut utiliser :

itemindex = np.where(array == 5)
print(itemindex[0][0], itemindex[1][0])  # Output: 1 1

Cela indique que la valeur 5 est située à l'index de ligne 1 et à l'index de colonne 1 dans le tableau, ce qui peut être confirmé par :

print(array[itemindex[0][0]][itemindex[1][0]])  # Output: 5

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!