如何在xarray中获取season作为JJAS而不是JJA？

Question

如何在计算季节平均值而不是季风月数组中获得季风月数( JJA )？

import xarray as xr
fname='/home/krishna/SIMS/P1/*.nc'
ds=xr.open_mfdataset(fname)
temp=ds.temp
sea_temp=temp.groupby('time.season').mean(dim='time')
sea_temp.plot.imshow(col='season',robust=True)

输出显示为JJA月份，而不是JJAS。如何计算JJAS的季节平均值(6-9月)

Answer 1

我可以通过如下所示编写一个函数来解决这个问题

def is_jjas(month):
    return (month >= 6) & (month <= 9)

seasonal_temp = temp.sel(time=is_jjas(temp['time.month']))

我希望这能帮助那些对印度夏季季风季(JJAS)做季节均值有同样疑问的人。