numpy.lib.Arrayterator

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

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

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

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

Параметры:

vararray_like

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

buf_sizeint, необязательно

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

См. также

numpy.ndenumerate

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

numpy.flatiter

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

numpy.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.