Функция Flatiter Python NumPy: перебор элементов массива

Функция numpy.flatiterв Python предоставляет итератор, который позволяет вам перебирать каждый элемент массива, как если бы это был плоский одномерный массив. Вот пример, демонстрирующий его использование:

import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
flat_iter = np.nditer(arr)
for element in flat_iter:
    print(element, end=' ')
# Output: 1 2 3 4 5 6

В этом примере мы создаем двумерный массив NumPy arrс формой (2, 3). Используя np.nditer()с arrв качестве входного аргумента, мы получаем плоский итератор flat_iter. Затем мы перебираем flat_iter, чтобы напечатать каждый элемент arrв плоском виде.

Вот еще несколько полезных методов, связанных с numpy.flatiter:

  1. iterindex: возвращает текущий индекс итератора.
  2. iternext: перемещает итератор к следующему элементу.
  3. finished: возвращает логическое значение, указывающее, достиг ли итератор конца.