import os, sys from importlib import import_module import scipy as sp import matplotlib.pyplot as pl from matplotlib import ticker, mlab, colors from matplotlib.cm import jet from mpl_toolkits.basemap.cm import sstanom import g5lib.plotters as ptrs import g5lib.domain as domain from g5lib import cmaps as g5cmaps from g5lib import g5dset varname = 'T' # Read variable exp = g5dset.read_exp(sys.argv[1]) exp.ctl = g5dset.Ctl(exp, 'geosgcm_ocn3d') if exp.ctl.grid['lev'][-1] < 0.0: exp.ctl.grid['lev'][:] *= -1 dd = exp.ctl.domain dd['dates'] = exp.dates ind = domain.Domain(**dd)(exp.ctl.grid, exp.ctl.time) exp.am = exp.ctl.tave(varname, ind['tind']) exp.am.data -= 273.16 exp.am.shiftgrid(30.) exp.lat_depth = exp.am.ave(3) exp.eq_depth = exp.am(lats=(-2.1, 2.1)).ave(2)
#!/usr/bin/env python import matplotlib matplotlib.use('Agg') import os, sys import scipy as sp import matplotlib.pyplot as pl from matplotlib.cm import coolwarm as g5cmap from g5lib import plotters, g5dset # Read data exp=g5dset.read_exp(sys.argv[1]) exp.ctl=g5dset.CtlMOM(exp) exp.mld=exp.ctl('mld',dates=exp.dates,levs=(0,)).clim(12) ind=[1,2,3] exp.mld.jfm=exp.mld.subset(tind=ind).mean(0); exp.mld.jfm.name+=', JFM' ind=[7,8,9] exp.mld.jas=exp.mld.subset(tind=ind).mean(0); exp.mld.jas.name+=', JAS' # Plots path=exp.plot_path try: os.makedirs(path) except OSError: pass clevs=sp.array([5, 10, 20, 50, 100, 150, 200, 300])