python xarray concat groupby datetime64 维度

Question

我有一个 xarray 数据集是：

ds
<xarray.Dataset>    
Dimensions:  (lat: 360, lon: 720, time: 3652)
Coordinates:
  * lon      (lon) float32 -179.75 -179.25 -178.75 -178.25 -177.75 -177.25     ...
  * lat      (lat) float32 89.75 89.25 88.75 88.25 87.75 87.25 86.75 86.25 ...
* time     (time) datetime64[ns] 2010-01-01 2010-01-02 2010-01-03 ...
Data variables:
dis    (time, lat, lon) float64 nan nan nan nan nan nan nan nan nan...

dis变量中有nans但整个数组不是nans。维度时间的长度对应10年的日数据（3652天）。

我想要做的是获取 10 年时间序列的每月平均值，每个月和每个网格正方形 (lat,lon)。所以输出数据集将是：

Dimensions:  (lat: 360, lon: 720, time: 12)  #<<< or 'months'

我看到的一个选项几乎可以满足我的要求：

ds.dis.groupby('time.month').mean()

然而，它的输出只是一个包含 12 项的数组。即我们失去了纬度和经度维度。

<xarray.DataArray 'dis' (month: 12)>
array([ 368.26764123,  394.0543304 ,  424.67056092,  476.94943773,
    522.383195  ,  516.37355647,  497.74700652,  472.46993274,
    456.87268206,  402.44729131,  367.41928436,  362.6121917 ])
Coordinates:
* month    (month) int64 1 2 3 4 5 6 7 8 9 10 11 12

我认为使用 datetime64 方法可能有一些简单的方法可以做到这一点，但我一直在努力充分理解它们。

唉，在写这篇文章的时候，我已经做到了：

stacked = xr.concat([ds.dis[tlist[month,:],:,:].mean(dim='time',skipna=True) for month in range(0,12)],dim='month')

给出：

<xarray.DataArray 'dis' (month: 12, lat: 360, lon: 720)>

但是，有没有另一种更pythonic的方式更符合使用groupby的第一行代码？

谢谢

Answer 1

为避免聚合每个子数组中的所有维度，您需要明确提供维度列表：

ds.dis.groupby('time.month').mean('time')

（有一次我们考虑将其设为 groupby 操作的默认行为，因为这通常是我们所期望的，但是不清楚如何触发当前对所有维度求和的默认行为。）

python xarray concat groupby datetime64 维度

python xarray concat groupby in datetime64 dimensions

python

datetime64

python-xarray