Как найти индекс значения в массиве numpy (с примерами)

Вы можете использовать следующие методы, чтобы найти позицию индекса определенных значений в массиве NumPy:

Метод 1: найти все позиции индекса стоимости

 n.p. where (x== value )

Способ 2. Найдите первую позицию индекса значения.

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

Способ 3. Найдите первую позицию индекса нескольких значений.

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

Следующие примеры показывают, как использовать каждый метод на практике.

Метод 1: найти все позиции индекса стоимости

Следующий код показывает, как найти каждую позицию индекса, равную определенному значению в массиве 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]),)

Из результата мы видим, что позиции индекса 4, 5 и 6 равны значению 8 .

Способ 2. Найдите первую позицию индекса значения.

Следующий код показывает, как найти первую позицию индекса, равную определенному значению в массиве 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

Из результата мы видим, что значение 8 сначала появляется в позиции индекса 4.

Способ 3. Найдите первую позицию индекса нескольких значений.

Следующий код показывает, как найти первую позицию индекса нескольких значений в массиве 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])

По результату мы видим:

• Значение 4 сначала появляется в позиции индекса 0.
• Значение 7 сначала появляется в позиции индекса 1.
• Значение 8 сначала появляется в позиции индекса 4.

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные операции в NumPy:

Как сопоставить функцию с массивом NumPy
Как преобразовать массив NumPy в список в Python
Как вычислить величину вектора с помощью NumPy