在Python xarray中将10年以上的季节性数据升采样为每日数据

时间:2019-03-05 11:54:26

标签: python numpy python-xarray

我有一个netCDF文件来获取季节性数据。加载到数据集后,它包含seasonlatitudelongitude维度。

print(dataset_seasonal_nc)


<xarray.Dataset>
Dimensions:               (latitude: 106, longitude: 193, season: 4)
Coordinates:
  * latitude              (latitude) float32 -39.2 -39.149525 ... -33.9
  * longitude             (longitude) float32 140.8 140.84792 ... 150.0
  * season                (season) object 'DJF' 'JJA' 'MAM' 'SON'
Data variables:
    FFDI 95TH PERCENTILE  (season, latitude, longitude) float64 dask.array<shape=(4, 106, 193), chunksize=(4, 106, 193)>

我需要将seasnonal数据上采样到10年的每日数据(例如,从1972年到1981年,总共3653天)。这意味着上采样的数据集对象应为:

<xarray.Dataset>
Dimensions:    (latitude: 106, longitude: 193, time: 3653)
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
  * time       (time) datetime64[ns] 1972-01-01T00:00:00 1972-01-02T00:00:00 1972-01-03T00:00:00 ... 1981-12-30T00:00:00 1981-12-31T00:00:00
Data variables:
    FFDI 95TH PERCENTILE  (time, latitude, longitude) float64 dask.array<shape=(3653, 106, 193), chunksize=(3653, 106, 193)>

一天的变量应与该天的季节的变量相同。这意味着1972-01-01、1972-02-02和1972-02-28的值应与DJF的季节;和1972-04-01、1972-05-02和1972-05-31应该具有与MAM季节相同的值。

我试图使用数据集的resample函数:

upsampled = dataset_seasonal_nc.resample(time='D').ffill()

但这给了我以下错误:

...\venv\lib\site-packages\xarray\core\dataset.py", line 896, in _construct_dataarray
    variable = self._variables[name]
KeyError: 'time'

1 个答案:

答案 0 :(得分:0)

这似乎是xarray's advanced label-based indexing的理想人选。我认为类似以下内容的方法应该起作用:

import pandas as pd

times = pd.date_range('1972', '1982', freq='D', closed='left')
time = xr.DataArray(times, [('time', times)])
upsampled = dataset_seasonal_nc.sel(season=time.dt.season)

这里time.dt.season是一个DataArray,代表与上采样数据集中的每次相关的季节标签:

In [16]: time.dt.season
Out[16]:
<xarray.DataArray 'season' (time: 3653)>
array(['DJF', 'DJF', 'DJF', ..., 'DJF', 'DJF', 'DJF'],
      dtype='|S3')
Coordinates:
  * time     (time) datetime64[ns] 1972-01-01 1972-01-02 1972-01-03 ...