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

Автор оригинала: Python Examples.

Итерация по элементам Numpy Array

Чтобы проиграть на небольшом массиве, вы можете использовать numpy.nditer объект итератора. NUMPY.NDITER предоставляет стандартный интерфейс iTerator Python для посещения каждого элемента в массиве Numpy. Любой размерный массив может быть итерацией.

Пример

В следующем примере у нас есть 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=' ')