nigsp.utils.prepare_ndim_iteration

nigsp.utils.prepare_ndim_iteration(data, idx)

Reshape data to have idx+1 dimensions.

This should allow iterations over unknown numbers of dimensions above idx when needed.

Parameters:

dataarray

The data to reiterate over.

idxint

The number of dimensions that should be fixed.

Returns:

array, array

The reshaped data and an empty array like it.