python - 将 xarray.apply(np.nansum) 与 args 一起使用

Question

我一直在尝试将 np.nansum 应用于 xr.Dataset (xarray)，但不断出现错误。对于 3D 数据集，我尝试应用 axis=2。语法不太清楚，我可能误解了 documentation ，但我已经尝试过:

ds.apply(np.nansum,axis=2)` and `ds.apply(lambda x: np.nansum(x,axis=2))

并得到相同的错误:

cannot set variable 'var' with 2-dimensional data without explicit dimension names. Pass a tuple of (dims, data) instead.

我猜这意味着它不知道要返回到新数据集对象的维度名称？任何想法如何解决这一问题？

有谁知道 xarray 为何以及何时实现 np.nansum() ？谢谢

Answer 1

您在这里遇到的问题是 nansum 返回一个 numpy ndarray，而不是 DataArray，这是函数的内容传入 apply 应该返回。

对于 nansum，您应该只使用 xarray.Dataset.sum，如果您的数据是 float ，它默认会跳过 NaN。