Як знайти індекс значення в масиві 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