微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

如何解决并行化沿 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 举报,一经查实,本站将立刻删除。