Como encontrar o índice de valor no array numpy (com exemplos)

Você pode usar os seguintes métodos para encontrar a posição do índice de valores específicos em uma matriz NumPy:

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

 n.p. where (x== value )

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

 n.p. where (x== value )[0][0]

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

 #define values of interest
vals = np. array ([ value1 , value2 , value3 ])

#find index location of first occurrence of each value of interest
exit = np. argsort (x)
go out[np. searchsorted (x, vals, sorter=sorter)]

Os exemplos a seguir mostram como usar cada método na prática.

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

O código a seguir mostra como encontrar cada posição de índice igual a um determinado valor em um array 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
n.p. where (x== 8 )

(array([4, 5, 6]),)

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

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

O código a seguir mostra como encontrar a primeira posição do índice igual a um determinado valor em um array 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
n.p. where (x== 8 )[0][0]

4

Pelo resultado, podemos ver que o valor 8 aparece primeiro na posição 4 do índice.

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

O código a seguir mostra como encontrar a primeira posição do índice de vários valores em um array 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
exit = np. argsort (x)
go out[np. searchsorted (x, vals, sorter=sorter)]

array([0, 1, 4])

Pelo resultado podemos ver:

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

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns no NumPy:

Como mapear uma função para um array NumPy
Como converter um array NumPy em uma lista em Python
Como calcular a magnitude de um vetor usando NumPy