数据集总和：在维度的开始和结束处，NaN而不是零

Question

我有一个数据集，其变量var_a取决于time和co。我想计算维度时间内的累计总和。如果在一个变量的开头/结尾有连续的NaN，我想排除它们，类似于在熊猫中应用first_valid / last_valid方法。在xarray中有解决方案吗？

一个最小的示例如下：

ds = xr.Dataset({'var_a': (['time', 'co'], np.array([[1,2,3], [np.nan, np.nan, 3]]).T)}, coords={'co': (['co', ], [1,2])})
ds_cs = ds.cumsum(dim='time')

我想要的图形：