dask.array.Array.compute_chunk_sizes

dask.array.Array.compute_chunk_sizes

Array.compute_chunk_sizes()[源代码]

计算Dask数组的块大小。这在块大小未知时特别有用(例如,当使用另一个Dask数组对一个Dask数组进行索引时)。

注释

此函数原地修改 Dask 数组。

示例

>>> 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),)