并行化沿 xarray 数据集维度工作的自定义函数

问题描述

我有一个函数 aggregate_xr_dataset(),它采用 4 个维度的 xarray - regionsyxtime 和 2 个变量 - { {1}} 和 ts_var 并聚合每个区域内的数据(沿 values_var 维度)。生成的 xarray 数据集有 3 个维度 - regionstimeregions

下面显示代码函数内部所做的更简单的版本,但输入和输出是相同的。

我想并行化这个函数,使其在每个区域并行工作,并将结果写入一个新的 xarray 数据集。我已经研究了 types,但无法弄清楚如何将它用于我的案例。 xr.apply_ufunc() 可以吗?

xr.apply_ufunc()

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)