dask.array.Array.compute_chunk_sizes — Dask documentation (original) (raw)

Compute the chunk sizes for a Dask array. This is especially useful when the chunk sizes are unknown (e.g., when indexing one Dask array with another).

This function modifies the Dask array in-place.

import dask.array as da import numpy as np x = da.from_array([-2, -1, 0, 1, 2], chunks=2) x.chunks ((2, 2, 1),) y = x[x <= 0] y.chunks ((nan, nan, nan),) y.compute_chunk_sizes() # in-place computation dask.array<getitem, shape=(3,), dtype=int64, chunksize=(2,), chunktype=numpy.ndarray> y.chunks ((2, 1, 0),)