Pode utilizar os seguintes métodos para encontrar a posição do índice de valores específicos numa matriz NumPy:

Método 1: Encontrar todas as posições de valor do índice

 np. where (x== valor ) 

Método 2: Encontrar a primeira posição de índice do valor

 np. where (x== valor ) 

Método 3: Encontrar a primeira posição de índice de vários valores

 #definir valores de interesse vals = np. array ([ valor1 , valor2 , valor3 ]) #encontrar a localização do índice da primeira ocorrência de cada valor de interesse sorter = np. argsort (x) sorter[np. searchsorted (x, vals, sorter=sorter)] 

Os exemplos seguintes mostram como utilizar cada método na prática.

Método 1: Encontrar todas as posições de valor do índice

O código seguinte mostra como encontrar todas as posições de índice que são iguais a um determinado valor numa matriz NumPy:

 import numpy as np #define array of values x = np. array ([4, 7, 7, 7, 8, 8, 8]) #find all index positions where x is equal to 8 np. where (x== 8 ) (array([4, 5, 6]),) 

A partir da saída, podemos ver que as posições de índice 4, 5 e 6 são todas iguais ao valor 8 .

Método 2: Encontrar a primeira posição de índice do valor

O código seguinte mostra como encontrar a primeira posição de índice que é igual a um determinado valor numa matriz NumPy:

 import numpy as np #define array of values x = np. array ([4, 7, 7, 7, 8, 8, 8]) #find first index position where x is equal to 8 np. where (x== 8 ) 4 

A partir do resultado, podemos ver que o valor 8 ocorre pela primeira vez na posição de índice 4.

Método 3: Encontrar a primeira posição de índice de vários valores

O código seguinte mostra como encontrar a primeira posição de índice de vários valores numa matriz NumPy:

 import numpy as np #define array of values x = np. array ([4, 7, 7, 7, 8, 8, 8]) #define values of interest vals = np. array ([4, 7, 8]) #find index location of first occurrence of each value of interest sorter = np. argsort (x) sorter[np. searchsorted (x, vals, sorter=sorter)] array([0, 1, 4]) 

A partir do resultado, podemos ver:

  • O valor 4 ocorre pela primeira vez na posição de índice 0.
  • O valor 7 ocorre pela primeira vez na posição de índice 1.
  • O valor 8 ocorre pela primeira vez na posição de índice 4.