>> 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)
Now we can iterate over ``a_itor``, and it will return arrays of size
two. Since `buf_size` was smaller than any dimension, the first
dimension will be iterated over first:
>>> for subarr in a_itor:
... if not subarr.all():
... print(subarr, subarr.shape) # doctest: +SKIP
>>> # [[[[0 1]]]] (1, 1, 1, 2)
Nc || _ || _ d |j D | _ d |j D | _ d |j D | _ d S )Nc g | ]}d S r .0dims M/opt/cloudlinux/venv/lib64/python3.11/site-packages/numpy/lib/arrayterator.py
z)Arrayterator.__init__..X s ---Ca--- c g | ]}|S r
r
r s r r z)Arrayterator.__init__..Y s ...SS...r c g | ]}d S ) r
r s r r z)Arrayterator.__init__..Z s ,,,3Q,,,r )varbuf_sizeshapestartstopstep)selfr r s r __init__zArrayterator.__init__T sY
--39---
..CI... ,,#),,, r c , t | j | S N)getattrr )r attrs r __getattr__zArrayterator.__getattr__\ s tx&&&r c t |t s|f}g }t | | j }}|D ]}|t u r<| t
d g||z
dz z t | }Gt |t r(| t
||dz d | | t | }t | |k r&|t
d f|t | z
z z
}| | j
| j }t t | j | j | j | D ]l\ }\ }} }
}||j pdz |j |< |
|j pdz |j |< ||j p| |z
z |j |< t# | |j | |j |< m|S )z-
Return a new arrayterator.
Nr r )
isinstancetuplelenndimEllipsisextendsliceintappend __class__r r enumeratezipr r r min) r indexfixedlengthdimsslice_outir r r s r __getitem__zArrayterator.__getitem___ s %'' HE5zz49 % %F!! eDkk]d6k!m <=== UFC((
% U66!8Q 7 78888 V$$$$e u:: eDkk^tCJJ7 7E nnTXt}55.7DJ 49e<</> /> 1 1 *A*tT6 FL$5A6CIaL &+"23CHQK 6;#<$u*=CHQK dCHQK00CHQKK
r c t d t | j | j | j D }| j | S )z-
Return corresponding data.
c 3 ( K | ]
}t | V d S r r( r ts r z)Arrayterator.__array__.. s6 3 3Quay 3 3 3 3 3 3r )r# r- r r r r )r r3 s r __array__zArrayterator.__array__ sR
3 3#
DIty+2 +2 3 3 3 3 3x r c # . K | D ]}|j E d{V dS )aG
A 1-D flat iterator for Arrayterator objects.
This iterator returns elements of the array to be iterated over in
`Arrayterator` one by one. It is similar to `flatiter`.
See Also
--------
Arrayterator
flatiter
Examples
--------
>>> a = np.arange(3 * 4 * 5 * 6).reshape(3, 4, 5, 6)
>>> a_itor = np.lib.Arrayterator(a, 2)
>>> for subarr in a_itor.flat:
... if not subarr:
... print(subarr, type(subarr))
...
0
N)flat)r blocks r r? zArrayterator.flat s> 2 " "E z ! ! ! ! ! ! ! ! " "r c p t d t | j | j | j D S )zk
The shape of the array to be iterated over.
For an example, see `Arrayterator`.
c 3 : K | ]\ }}}||z
d z
|z d z V dS )r Nr
)r r r r s r r< z%Arrayterator.shape.. sK 7 72C%ttEz!|d*1, 7 7 7 7 7 7r )r# r- r r r )r s r r zArrayterator.shape sD 7 7DJ 49557 7 7 7 7 7r c # K d | j D rd S | j d d }| j d d }| j d d }| j j } | j pt t | j }d}t |dz
dd D ]}|dk r|| dz ||< n;|| j | k r|| ||| z z ||< |}n| j | ||< t | j | || ||< || j | z }t d t ||| D }| j | V || ||< t |dz
dd D ]J}|| | j | k r1| j | ||< ||dz
xx | j |dz
z
cc< K|d | j d k rd S z)Nc g | ]
}|d k | S r r
r s r r z)Arrayterator.__iter__.. s 2 2 2CCr Tr r c 3 ( K | ]
}t | V d S r r9 r: s r r< z(Arrayterator.__iter__.. s& EE5!9EEEEEEr )
r r r r r r% r r r ranger. r# r- ) r r r r ndimscountrundimr5 r3 s r __iter__zArrayterator.__iter__ s 2 24: 2 2 2 F
111
y|y|
" MrT sy
K K K K K K K K K Kr