dask.array.Array.partitions

dask.array.Array.partitions

property Array.partitions

按分区切片数组。dask 数组 .blocks 属性的别名。

此别名允许您编写与 dask 数组和 dask 数据帧都兼容的代码。

这将返回一个 Blockview 对象,该对象提供了一个类似数组的接口来访问 dask 数组的块。对 Blockview 对象进行类似 Numpy 的索引操作将返回一个由选定块组成的新 dask 数组。

你可以像索引形状等于每个维度中块数的 numpy 数组一样索引 array.blocks``(可用作 array.blocks.size)。输出数组的维度与该数组的维度匹配,即使传递的是整数索引。不支持使用 ``np.newaxis 或多个列表进行切片。

返回
da.array.Blockview 的一个实例

示例

>>> import dask.array as da
>>> x = da.arange(8, chunks=2)
>>> x.partitions.shape # aliases x.numblocks
(4,)
>>> x.partitions[0].compute()
array([0, 1])
>>> x.partitions[:3].compute()
array([0, 1, 2, 3, 4, 5])
>>> x.partitions[::2].compute()
array([0, 1, 4, 5])
>>> x.partitions[[-1, 0]].compute()
array([6, 7, 0, 1])
>>> x.partitions.ravel() 
[dask.array<blocks, shape=(2,), dtype=int64, chunksize=(2,), chunktype=numpy.ndarray>,
 dask.array<blocks, shape=(2,), dtype=int64, chunksize=(2,), chunktype=numpy.ndarray>,
 dask.array<blocks, shape=(2,), dtype=int64, chunksize=(2,), chunktype=numpy.ndarray>,
 dask.array<blocks, shape=(2,), dtype=int64, chunksize=(2,), chunktype=numpy.ndarray>]