numpy.lib.Arrayterator

class numpy.lib.Arrayterator(var, buf_size=None) [source]

Буферизованный итератор для больших массивов.

Arrayterator создает буферизованный итератор для чтения больших массивов небольшими смежными блоками. Класс полезен для объектов, хранящихся в файловой системе. Он позволяет итерироваться по объекту без чтения всего в память; вместо этого небольшие блоки считываются и итерируются.

Arrayterator может использоваться с любым объектом, поддерживающим многомерные срезы. Это включает массивы NumPy, а также переменные из Scientific.IO.NetCDF или pynetcdf, например.

Параметры

vararray_like

Объект для итерирования.

buf_sizeint, optional

Размер буфера. Если buf_size указан, максимальный объем данных, который будет загружен в память, составляет buf_size элементов. По умолчанию None, что будет читать столько элементов, сколько возможно в память.

См. также

ndenumerate

Многомерный итератор массивов.

flatiter

Итератор плоского массива.

memmap

Создание сопоставления памяти с массивом, хранящимся в двоичном файле на диске.

Примечания

Алгоритм работает, сначала определяя «рабочую размерность», по которой будут извлекаться блоки. Учитывая массив с размерностями (d1, d2, ..., dn), например, если buf_size меньше, чем d1, будет использована первая размерность. Если, с другой стороны, d1 < buf_size < d1*d2, будет использована вторая размерность и так далее. Блоки извлекаются по этой размерности, и когда возвращается последний блок, процесс продолжается с следующей размерностью, пока все элементы не будут прочитаны.

Примеры

>>> a = np.arange(3 * 4 * 5 * 6).reshape(3, 4, 5, 6)
>>> a_itor = np.lib.Arrayterator(a, 2)
>>> a_itor.shape
(3, 4, 5, 6)

Теперь мы можем итерироваться по a_itor, и это вернет массивы размера два. Так как buf_size было меньше любой размерности, сначала будет итерироваться по первой размерности:

>>> for subarr in a_itor:
...     if not subarr.all():
...         print(subarr, subarr.shape) 
>>> # [[[[0 1]]]] (1, 1, 1, 2)

Атрибуты

var

buf_size

start

stop

step

shape

Форма массива, по которому происходит итерирование.

flat

1-мерный итератор для объектов Arrayterator.