In [1]:
%matplotlib inline
from matplotlib import pyplot as plt
In [2]:
import os, sys
import numpy as np
from numpy import ma
import xray
In [3]:
dpath = os.path.join(os.environ['HOME'], 'data/NCEP1')
In [ ]:
dset_u = xray.open_dataset(os.path.join(dpath, 'wind/uwnd.mon.mean.nc'))
dset_v = xray.open_dataset(os.path.join(dpath, 'wind/vwnd.mon.mean.nc'))
In [ ]:
dset_u = dset_u.sel(time=slice('1948','2014'))
dset_v = dset_v.sel(time=slice('1948','2014'))
In [ ]:
lat = dset_u['lat'].data
lon = dset_u['lon'].data
In [ ]:
dates = dset_u['time'].data
In [ ]:
uwnd_1000 = dset_u.sel(level=1000)['uwnd'].data
In [ ]:
vwnd_1000 = dset_v.sel(level=1000)['vwnd'].data
In [ ]:
uwnd_850 = dset_u.sel(level=850)['uwnd'].data
In [ ]:
vwnd_850 = dset_v.sel(level=850)['vwnd'].data
In [ ]:
uwnd_200 = dset_u.sel(level=200)['uwnd'].data
In [ ]:
vwnd_200 = dset_v.sel(level=200)['vwnd'].data
In [ ]:
d = {}
In [ ]:
d['time'] = ('time',dates)
d['latitudes'] = ('latitudes',lat)
d['longitudes'] = ('longitudes', lon)
d['uwnd_1000'] = (['time','latitudes','longitudes'], uwnd_1000)
d['vwnd_1000'] = (['time','latitudes','longitudes'], vwnd_1000)
d['uwnd_850'] = (['time','latitudes','longitudes'], uwnd_850)
d['vwnd_850'] = (['time','latitudes','longitudes'], vwnd_850)
d['uwnd_200'] = (['time','latitudes','longitudes'], uwnd_200)
d['vwnd_200'] = (['time','latitudes','longitudes'], vwnd_200)
In [ ]:
wind = xray.Dataset(d)
In [ ]:
wind
In [ ]:
wind.to_netcdf('./outputs/NCEP1_monthly_wind_1948_2014.nc')
In [ ]:
dset_u.close()
dset_v.close()
In [4]:
dset_hgt = xray.open_dataset(os.path.join(dpath, 'hgt/hgt.mon.mean.nc'))
In [5]:
dset_hgt
Out[5]:
In [6]:
lat = dset_hgt['lat'].data
lon = dset_hgt['lon'].data
In [7]:
dset_hgt = dset_hgt.sel(time=slice('1948','2014'))
In [8]:
dates = dset_hgt['time'].data
In [9]:
hgt_1000 = dset_hgt.sel(level=1000)['hgt'].data
In [10]:
hgt_850 = dset_hgt.sel(level=850)['hgt'].data
In [11]:
hgt_200 = dset_hgt.sel(level=200)['hgt'].data
In [12]:
d = {}
d['time'] = ('time',dates)
d['latitudes'] = ('latitudes',lat)
d['longitudes'] = ('longitudes', lon)
d['hgt_1000'] = (['time','latitudes','longitudes'], hgt_1000)
d['hgt_850'] = (['time','latitudes','longitudes'], hgt_850)
d['hgt_200'] = (['time','latitudes','longitudes'], hgt_200)
In [13]:
hgt = xray.Dataset(d)
In [14]:
hgt
Out[14]:
In [15]:
hgt.to_netcdf('./outputs/NCEP1_monthly_hgt_1948_2014.nc')
In [17]:
!cp ./outputs/NCEP1_monthly_hgt_1948_2014.nc /Users/nicolasf/research/NIWA/paleo/pict/data/NCEP1_monthly_hgt_1948_2014.nc
In [16]:
dset_hgt.close()
In [ ]:
dset_omega = xray.open_dataset(os.path.join(dpath, 'omega/omega.mon.mean.nc'))
In [ ]:
dset_omega
In [ ]:
dset_omega = dset_omega.sel(level=500, time=slice('1948','2014'))
In [ ]:
lat = dset_omega['lat'].data
lon = dset_omega['lon'].data
In [ ]:
dates = dset_omega['time'].data
In [ ]:
omega_500 = dset_omega['omega'].data
In [ ]:
d = {}
d['time'] = ('time',dates)
d['latitudes'] = ('latitudes',lat)
d['longitudes'] = ('longitudes', lon)
d['omega_500'] = (['time','latitudes','longitudes'], omega_500)
In [ ]:
omega = xray.Dataset(d)
In [ ]:
omega.to_netcdf('./outputs/NCEP1_monthly_omega_1948_2014.nc')
In [ ]:
omega.close()
In [ ]:
omega
In [ ]:
In [ ]: