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

Trouver le premier index d'un élément dans un tableau NumPy

Dans les listes Python, la méthode index() nous permet de récupérer le premier occurrence d'un élément. NumPy peut-il fournir des fonctionnalités similaires pour les tableaux ?

Réponse

Oui, NumPy offre une méthode pratique pour localiser le premier index d'un élément dans un tableau. Vous pouvez utiliser la fonction np.where comme suit :

import numpy as np

array = np.array([1, 2, 3])
item = 2

itemindex = np.where(array == item)

La fonction np.where renvoie un tuple contenant les indices de ligne et de colonne où se trouve l'élément.

Exemple

Si votre tableau est bidimensionnel et contient l'élément à deux emplacements, puis :

array[itemindex[0][0]][itemindex[1][0]]  # equals item
array[itemindex[0][1]][itemindex[1][1]]  # also equals item

Cela permet d'accéder facilement à l'élément souhaité en fonction de sa première occurrence.

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!