У меня есть набор данных с переменной 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')
Что я хочу графически:
