python - 将 DataArray 合并到 xarray 数据集中

Question

是否有比 xarray 数据集中所有 DataArray 更好的方法来求和

sum(d for d in ds.data_vars.values())

这可行，但看起来有点笨拙。是否有相当于对 pandas DataFrame 列进行求和的方法？

请注意，ds.sum() 方法适用于每个 DataArray - 但我想组合 DataArray。

Answer 1

我假设您还想对每个数据变量求和，例如 sum(d.sum() for d in ds.data_vars.values())。在 xarray 的 future 版本(v0.10 中还没有)中，这将更加简洁:您将能够编写 sum(d.sum() for d in ds.values())。

另一种选择是将数据集转换为单个 DataArray 并立即求和，例如，ds.to_array().sum()。如果您有不同维度的数据变量，这会降低效率。