enumerate

caput.mpiarray.MPIArray.enumerate(axis: int) → collections.abc.Iterator[tuple[int, int]]

Helper for enumerating over a given axis.

Parameters:

axisint

Axis to enumerate over.

Returns:

enumeratorenumerate

An enumerator which returns the local index into the array and the global index it corresponds to.