在 xarray 数据集中组合 DataArray

Question

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

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

这有效，但似乎有点笨拙。是否有等价于对 Pandas DataFrame 列求和？

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

Answer 1

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

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