flox.rechunk_for_blockwise#
- flox.rechunk_for_blockwise(array, axis, labels)[source]#
Rechunks array so that group boundaries line up with chunk boundaries, allowing embarrassingly parallel group reductions.
This only works when the groups are sequential (e.g. labels =
[0,0,0,1,1,1,1,2,2]). Such patterns occur when using.resample.- Parameters:
- arrayDaskArray
Array to rechunk
- axisint
Axis along which to rechunk the array.
- labelsnp.ndarray
Group labels
- Returns:
- DaskArray
Rechunked array