I have an numpy array: a[i_x, i_y, i_z], where 0 \\leq i_x < nx, 0 \\leq i_y < ny, 0 \\leq i_z; what\'s the numpy mgrid that corresponds to this -- is it
a[i_x, i_y, i_z]