如何调用xarray的groupby函数按年和月的组合对数据进行分组

Question

我有一个 DataArray 对象，用于跨越几年的日常数据集。这有一个变量和三个维度，分别命名为latitude,longitude和time(daily)。时间坐标就像time (time) datetime64[ns] 2016-01-01 2016-01-02 ... 2018-12-31

我想通过 DataArray 的 groupby 函数按年和月的组合对数据进行分组。但是下面的代码只给了我 中的时间坐标int64，比如 1, 2, 3, ..., 12。

da_groupby_monthly = da.groupby('time.month').sum('time')
print(da_groupby_monthly)

输出：

<xarray.DataArray (month: 12, latitude: 106, longitude: 193)>
dask.array<shape=(12, 106, 193), dtype=int32, chunksize=(1, 106, 193)>
Coordinates:
  * latitude   (latitude) float32 -39.2 -39.149525 ... -33.950478 -33.9
  * longitude  (longitude) float32 140.8 140.84792 140.89584 ... 149.95209 150.0
  * month      (month) int64 1 2 3 4 5 6 7 8 9 10 11 12

如何保持时间的数据类型datetime64[ns]并使月份坐标为“2016-01”、“2016-02”、“2016-03”、……、“2018-12”等在。

Answer 1

要更一般地对多个变量（例如年和月）执行 xarray groupby 操作，您可以在 pandas MultiIndex 中组合变量，使其成为非维度坐标，并将其传递给 groupby：

import pandas as pd
year_month_idx = pd.MultiIndex.from_arrays([da['time.year'], da['time.month']])
da.coords['year_month'] = ('time', year_month_idx)
da_monthly = da.groupby('year_month').sum()

您还可以groupby通过堆叠坐标创建一个 MultiIndex 以供使用。例如，给定一组纬度/经度坐标，您可以按所有唯一的经纬度位置分组：

da_stacked = da.stack(latlon=['lat','lon'])
da_stacked.groupby('latlon').sum()

Answer 2

我喜欢使用重采样方法。尝试这个：

da_monthly  = da.resample('1m', on='time').sum()