The array_split package provides the means to partition an array (or array shape) using any of the following criteria: Per-axis indices indicating the cut positions. Per-axis number of sub-arrays.
The following quickstart.py script creates a zero-initialised array and performs some element assignments: import mpi_array as mpia # creates zero-initialized PGAS (distributed) array dary = ...