Jak znaleźć indeks wartości w tablicy numpy (z przykładami)

Aby znaleźć pozycję indeksu określonych wartości w tablicy NumPy, możesz użyć następujących metod:

Metoda 1: Znajdź wszystkie pozycje indeksu wartości

 n.p. where (x== value )

Metoda 2: Znajdź pozycję indeksu pierwszej wartości

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

Metoda 3: Znajdź pierwszą pozycję indeksu wielu wartości

 #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)]

Poniższe przykłady pokazują, jak zastosować każdą metodę w praktyce.

Metoda 1: Znajdź wszystkie pozycje indeksu wartości

Poniższy kod pokazuje, jak znaleźć każdą pozycję indeksu równą określonej wartości w tablicy 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]),)

Z wyniku widzimy, że pozycje indeksu 4, 5 i 6 są równe wartości 8 .

Metoda 2: Znajdź pozycję indeksu pierwszej wartości

Poniższy kod pokazuje, jak znaleźć pierwszą pozycję indeksu równą określonej wartości w tablicy 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

Z wyniku widzimy, że wartość 8 pojawia się najpierw na pozycji indeksu 4.

Metoda 3: Znajdź pierwszą pozycję indeksu wielu wartości

Poniższy kod pokazuje, jak znaleźć pierwszą pozycję indeksu wielu wartości w tablicy 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])

Z wyniku możemy zobaczyć:

• Wartość 4 pojawia się najpierw na pozycji indeksu 0.
• Wartość 7 pojawia się najpierw na pozycji indeksu 1.
• Wartość 8 pojawia się najpierw na pozycji indeksu 4.

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak wykonywać inne typowe operacje w NumPy:

Jak zmapować funkcję na tablicę NumPy
Jak przekonwertować tablicę NumPy na listę w Pythonie
Jak obliczyć wielkość wektora za pomocą NumPy