Python 指定如何对dask数据帧进行分区？

我有一个由

id

和

date

索引的索引。我想使用dask对每个id并行运行一些回归。我知道dask将df拆分为N个分区，但是有没有办法强制它按

id

列进行拆分？这样，当我映射分区时，我可以简单地将滚动回归函数应用于每个分区