我有一个 xarray DataArray,我想为整个时间序列选择四月、五月、六月(类似于 time.season=='JJA')。
它的结构如下:
<xarray.DataArray 't2m' (time: 492, latitude: 81, longitude: 141)>
我之前通过以下方式选择了 JJA:
seasonal_data =temp_data.sel(time=temp_data['time.season']=='JJA')
我想做同样的事情,但用月份“AMJ”代替。我可以添加任何我可能遗漏的细节。
谢谢
最佳答案
选择自定义月份的最简单方法是使用 bool 掩码,例如,
def is_amj(month):
return (month >= 4) & (month <= 6)
seasonal_data = temp_data.sel(time=is_amj(temp_data['time.month']))
请注意,您需要使用位运算符,如 &
或 |
因为 Python 的内置 and
和 or
不适用于向量。此外,您需要括号,因为按位运算符的优先级高于比较。
关于python - 根据特定月份选择 xarray/pandas 索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40272222/