Итерация по всем элементам NumPy Array в Python

Чтобы перебрать массив NumPy в Python, вы можете использовать объект-итератор numpy.nditer, который предоставляет стандартный интерфейс Iterator для посещения каждого элемента. Можно повторять любой размерный массив.

Пример

В следующем примере у нас есть 2D-массив, и мы используем numpy.nditer для печати всех элементов массива.

import numpy as np

#2D array
a = (np.arange(8)*2).reshape(2,4)

#print array
print("The array\n",a)

print("\nIterating over all the elemnets of array")
#iterate over elements of the array
for x in np.nditer(a):
	print(x, end=' ')