Плоский перебор в NumPy

Атрибут flat даёт итератор по всем элементам массива в порядке обхода. Им удобно пользоваться, когда нужен один индекс на всю таблицу.

Выведем элементы таблицы через flat в цикле:

import numpy as np

table = np.array([[1, 2], [3, 4]])
for cell in table.flat:
	print(cell, end=' ')
print() # выведет 1 2 3 4

Через flat можно и менять ячейки. Запись в элемент итератора меняет исходный массив:

import numpy as np

table = np.array([[1, 2], [3, 4]])
table.flat[0] = 99
print(table) # выведет [[99  2] [ 3  4]]

Создайте таблицу [[5, 6], [7, 8]] и выведите первый элемент плоского обхода.

Создайте таблицу из двух строк [1, 2] и [3, 4], запишите 0 в последнюю ячейку плоского обхода и выведите таблицу.