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-D итератор для объектов Arrayterator.