caput.mpiarray.MPIArray.enumerate

MPIArray.enumerate(axis)[source]

Helper for enumerating over a given axis.

Parameters:

axis : integer

Which access to enumerate over.

Returns:

iterator : (local_index, global_index)

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