xarray：将数据变量与新的连续维度中的离散观察值相结合答案

【问题标题】：xarray: Combine data variables with discrete observations in a new continuous dimensionxarray：将数据变量与新的连续维度中的离散观察值相结合
【发布时间】：2023-01-24 04:40:11
【问题描述】：

我正在使用一个作物日历来记录给定物候状态发生的一年中的哪一天（doy）——这里是平均种植（plant）和收获（harvest）季节（注意nan打印下面是海洋上的像素，其他值包含int）：

<xarray.Dataset>
Dimensions:  (y: 2160, x: 4320)
Coordinates:
  * x        (x) float64 -180.0 -179.9 -179.8 -179.7 ... 179.7 179.8 179.9 180.0
  * y        (y) float64 89.96 89.88 89.79 89.71 ... -89.71 -89.79 -89.88 -89.96
Data variables:
    plant    (y, x) float32 nan nan nan nan nan nan ... nan nan nan nan nan nan
    harvest  (y, x) float32 nan nan nan nan nan nan ... nan nan nan nan nan nan

我需要将这两个变量组合在维度数据数组中 (doy: 365, y: 2160, x: 4320) 以便跟踪每个像素的物候状态作为 doy 的函数。从概念上讲，到目前为止我确定的步骤是：

为每个状态分配一个数值，例如off=0、plant=1、harvest=2
在新dataarray的doy维度中以doy作为对应日期的索引，赋值状态对应的数值
使用类似于 pandas.DataFrame.fillna 和 method='ffill' 的方式完成中间的值
我浏览了Reshaping and reorganizing data 和Combining Data 页面，但根据我目前对 xarray 的理解，我真的不知道从哪里开始。

谁能指出我的方向？我想做的事情是仅使用矩阵运算就可以实现，还是我必须引入循环？

PS：为问题本身的混乱表述道歉。我想这只反映了我仍然缺少的一些基本知识。

【问题讨论】：

标签： python python-xarray

【解决方案1】：

您可以利用 xarray 的自动广播规则创建一个布尔掩码，其中包含所有日期高于/低于由 x/y 索引的日期数组的日期：

# I'm assuming your "day of year" values are 1-indexed, and you're
# using a 365-day calendar. I'll leave leap year handling to you :)
days_of_year = xr.DataArray(
    np.arange(1, 366), dims=["day_of_year"], coords=[np.arange(1, 366)],
)

# broadcast against eachother with (x, y) <= (day_of_year, )
planted = ds.plant <= days_of_year
harvested = ds.harvest <= days_of_year

state = planted + harvested

结果将是一个维度为 (x, y, day_of_year) 的数组，其中包含您描述的代码。

关于生长期数据的说明：

我以前处理过此类数据，需要注意的一件事是全球范围内生长季节跨越 1 月 1 日的地区，这会破坏您的上述方法。或者，您可以只创建一个“is_growing_season”掩码，它可以灵活地处理具有跨越日历年的生长季节的区域：

is_growing_season = xr.where(
    ds.harvest >= ds.plant,
    ((days_of_year >= ds.plant) & (days_of_year <= ds.harvest)),
    ((days_of_year >= ds.plant) | (days_of_year <= ds.harvest)),
)

【讨论】：