NumPy — мощная библиотека Python для научных вычислений, обеспечивающая эффективные операции с массивами. При работе с массивами NumPy вам часто может потребоваться найти первое вхождение определенного элемента. В этой статье мы рассмотрим несколько методов решения этой задачи, а также приведем примеры кода.

Методы поиска первого вхождения в массиве NumPy:

1. Использование np.where():
   Функция np.where() возвращает индексы, в которых выполняется заданное условие. Передав желаемое значение в качестве условия, мы можем получить индекс его первого вхождения.

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 4, 3, 2, 1])
value = 4
indices = np.where(arr == value)[0]
if len(indices) > 0:
    first_index = indices[0]
    print("First occurrence index:", first_index)
else:
    print("Value not found in the array.")

2. Использование np.argmax():
   Функция np.argmax() возвращает индексы максимальных значений вдоль указанной оси. Применяя его к логическому массиву, созданному с помощью поэлементного сравнения, мы можем получить индекс первого вхождения.

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 4, 3, 2, 1])
value = 4
indices = np.argmax(arr == value)
if indices > 0:
    print("First occurrence index:", indices)
else:
    print("Value not found in the array.")

3. Использование np.nonzero():
   Функция np.nonzero() возвращает индексы ненулевых элементов в заданном массиве. Применяя его к логическому массиву, созданному с помощью поэлементного сравнения, мы можем получить индекс первого вхождения.

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 4, 3, 2, 1])
value = 4
indices = np.nonzero(arr == value)[0]
if len(indices) > 0:
    first_index = indices[0]
    print("First occurrence index:", first_index)
else:
    print("Value not found in the array.")

4. Использование np.flatnonzero():
   Функция np.flatnonzero() возвращает индексы ненулевых элементов в сплющенной версии входного массива. Применяя его к логическому массиву, созданному с помощью поэлементного сравнения, мы можем получить индекс первого вхождения.

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 4, 3, 2, 1])
value = 4
indices = np.flatnonzero(arr == value)
if len(indices) > 0:
    first_index = indices[0]
    print("First occurrence index:", first_index)
else:
    print("Value not found in the array.")