Python - Как сгладить сюжет Xarray? - PullRequest
Новая
       10

Python - Как сгладить сюжет Xarray?

0 голосов
/ 29 августа 2018

У меня есть несколько файлов NetCDF, из которых мне нужно построить некоторые переменные. Я использую следующий маленький код для построения нужных мне данных: 

import matplotlib.pyplot as plt
import matplotlib.dates as mdates
%matplotlib inline
import xarray as xr
os.chdir
fname = "/home/data.nc"
ds = xr.open_dataset(fname)
ds['Evap_obs'].plot()
plt.show()

, которые работают просто отлично и дают следующий рисунок:

enter image description here

Моя проблема сейчас в том, что этот сюжет не очень читабелен. Поэтому я хочу сгладить данные заранее. Для этого у меня есть эта маленькая функция: 

def mov_avg(x,window,min_count=None,axis=-1):
    import bottleneck as bn 
    yy=np.ma.filled(np.ma.fix_invalid(x),np.nan)
    yyF=np.ma.masked_all(yy.shape)
    xtmp=bn.move_mean(yy,window,min_count=min_count,axis=axis)
    wd1=(window-1)/2
    ndim = len(yy.shape)
   #print xtmp.shape,ndim,axis,window,wd1
    if ndim ==1 :
   #print wd1,-wd1,wd-1
        yyF[wd1:-wd1]=np.ma.fix_invalid(xtmp[window-1:])

    elif ndim == 2:
        if axis==-1 or axis==1:
            yyF[:,wd1:-wd1]=np.ma.fix_invalid(xtmp[:,window-1:])
        elif axis==0:
            yyF[wd1:-wd1,:]=np.ma.fix_invalid(xtmp[window-1:,:])

    return yyF

Затем, когда я хочу запустить эту функцию для моих данных, чтобы сгладить ее следующим образом (используя окно из 7 временных шагов для сглаживания): 

import matplotlib.pyplot as plt
import matplotlib.dates as mdates
%matplotlib inline
import xarray as xr
os.chdir
fname = "/home/data.nc"
ds = xr.open_dataset(fname)
obs = ds['Evap_obs']
obs = mov_avg(obs,7)
obs.plot()

Это не работает, и я получил следующее ValueError: 

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-18-6a8a2dd438c6> in <module>()
      7 #dsloc = ds.sel(lon=230.5,lat=55.0,method='nearest')
      8 obs = ds['Evap_obs']
----> 9 obs = mov_avg(obs,7)
     10 obs.plot()
     11 #ds['Evap_obs'].plot()

<ipython-input-11-c01b37a40c9a> in mov_avg(x, window, min_count, axis)
      4     yy=np.ma.filled(np.ma.fix_invalid(x),np.nan)
      5     yyF=np.ma.masked_all(yy.shape)
----> 6     xtmp=bn.move_mean(yy,window,min_count=min_count,axis=axis)
      7     wd1=(window-1)/2
      8     ndim = len(yy.shape)

ValueError: Moving window (=7) must between 1 and 1, inclusive

Кто-нибудь знает, как сгладить мои данные, пожалуйста?

Ответы [ 2 ]

0 голосов
/ 29 августа 2018

Xarray включает метод прокатки для этих видов операций. 

ds['Evap_obs'].rolling(time=7).mean().plot()

Документация xarray включает несколько примеров использования этого метода: http://xarray.pydata.org/en/stable/computation.html#rolling-window-operations

0 голосов
/ 29 августа 2018

Ниже приведена простая реализация скользящего среднего из этого ответа с использованием только numpy: 

import numpy as np
def moving_average(a, n=3) :
    ret = np.cumsum(a, dtype=float)
    ret[n:] = ret[n:] - ret[:-n]
    return ret[n - 1:] / n
...