как найти индекс элемента numpy

Аватар пользователя Ivan Mamtsev
Ivan Mamtsev
10 сентября 2024

Самым простым способом будем использование метода .where()

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 2])
indices = np.where(arr == 2)
print(indices[0])  # [1 5]

Как видно выше, метод выведет все совпадения с элементом.

0 0
Познакомьтесь с основами аналитики данных бесплатно