def __timeseriesa(data, cdata, variable, ncdf, cncdf, ax):
#construct strings
label = labels[variable]
#format data
data = ncdf.formatdata(data, 'timeseries', cncdf)
cdata = ncdf.formatdata(cdata, 'timeseries')
#create plot
x = calc.getx(ncdf)
y = calc.anomalize(data, cdata, ncdf, cncdf)
lp = plt.plot(x, y, 'k')
plt.xlim(x[0], x[-1])
#label & title
plt.xlabel('Years after eruption')
plt.ylabel(label)
#additional formatting
ax = plt.gca()
ax.yaxis.set_major_locator(ticker.MaxNLocator(nbins=5))
ax.ticklabel_format(axis='y', style='sci', scilimits=(-1,1))
ax.spines['bottom'].set_position('zero')
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')
return ax
def timeseriesa(data, cdata, variable, ncdf, cncdf):
#construct strings
label = labels[variable]
if variable in names:
title = names[variable]
else:
title = variable
subtitle = ncdf.filetitle + ' - ' + cncdf.filetitle
filename = (variable + '_' + filenames[ncdf.filename] + '_tsa_' +
filenames[cncdf.filename] + '.eps')
#create plot
x = calc.getx(ncdf)
y = calc.anomalize(data, cdata, ncdf, cncdf)
lp = plt.plot(x, y, 'k')
plt.xlim(x[0], x[-1])
#label & title
plt.xlabel('Years after eruption')
plt.ylabel(label)
ax.yaxis.ticklabel_format(style='sci', axis='y', scilimits=(-2,2))
ax.yaxis.locator_params(nbins=3, axis='y')
#additional formatting
ax = plt.gca()
plt.rc('font', **font)
plt.title(subtitle)
ax.spines['bottom'].set_position('zero')
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')
plt.savefig(filename)
plt.close()
def __prvtimea(data, cdata, variable, ncdf, cncdf, ax):
#format data
data = ncdf.formatdata(data, 'prvtime', cncdf)
cdata = ncdf.formatdata(cdata, 'prvtime')
#anomaly
data = calc.anomalize(data, cdata, ncdf, cncdf)
#construct strings
label = labels[variable]
#colormap
cmap = LinearSegmentedColormap('bwr', colors)
#calculate params
x, y = calc.getx(ncdf), ple
z = data.transpose()
datamax, datamin = calc.getmax(data, True), calc.getmin(data, True)
v = np.linspace(datamin, datamax, 41)
#create plot
cp = plt.contourf(x, y, z, v, extend='both', color=None, cmap=cmap,
vmin=datamin, vmax=datamax)
plt.yscale('log')
plt.xlim(x[0], x[-1])
plt.ylim(.1, 200)
#create cbar
ticks = np.linspace(datamin, datamax, 3)
cbar = plt.colorbar(cp, orientation='horizontal', extend='both',
aspect=40, ticks=ticks, format='%.2g')
cbar.ax.set_xlabel(label)
#label & title
plt.xlabel('Years after eruption')
plt.ylabel('Pressure (hPa)')
#additional formatting
ax = plt.gca()
ax.invert_yaxis()
#add control stats
cmean = stats.nanmean(cdata.flat)
#cmean = calc.anlmean(cdata)
cstd = stats.nanstd(cdata.flat)
#cstd = calc.anlstd(cdata)
string = ('Control mean: ' + '%.2g'%cmean +
'\nControl standard deviation: ' + '%.2g'%cstd)
plt.figtext(.02,.02,string, fontsize=12)
#show/save plot
return ax
def prvtimea(data, cdata, variable, ncdf, controlncdf):
fig = plt.figure(figsize=figsize)
ax = fig.add_subplot(111)
fig.subplots_adjust(left=.2)
#anomaly
data = calc.anomalize(data, cdata, ncdf, controlncdf)
#construct strings
label = labels[variable]
if variable in names:
title = names[variable]
else:
title = variable
if variable in variablefilenames:
variabletitle = variablefilenames[variable]
else:
variabletitle = variable
filename = (variabletitle + '_' + filenames[ncdf.filename] + '_pvta_' +
filenames[controlncdf.filename] + '.eps')
#colormap
cmap = LinearSegmentedColormap('bwr', colors)
#calculate params
x, y = calc.getx(ncdf), ple
z = data.transpose()
datamax, datamin = 0, 0
if ((filenames[ncdf.filename] != 'q150') or
(filenames[controlncdf.filename] != 'q0')):
try:
datamax = minmax[variable + 'a'][1]
datamin = minmax[variable + 'a'][0]
except KeyError:
datamax, datamin = calc.getmax(data, True), calc.getmin(data, True)
except Error:
print Error
else:
try:
datamax = minmax[variable + 'aa'][1]
datamin = minmax[variable + 'aa'][0]
except KeyError:
datamax, datamin = calc.getmax(data, True), calc.getmin(data, True)
except Error:
print Error
v = np.linspace(datamin, datamax, 17)
#create plot
cp = ax.contourf(x, y, z, v, extend='both', color=None, cmap=cmap,
vmin=datamin, vmax=datamax)
plt.yscale('log')
plt.xlim(x[0], x[-1])
plt.ylim(.1, 200)
#create cbar
cbar = None
ticks = np.linspace(datamin, datamax, 5, endpoint=True)
format = ScalarFormatter()
format.set_powerlimits((-2,2))
cbar = plt.colorbar(cp, orientation='horizontal',
aspect=15, ticks=ticks, format=format,
pad = .2)
cbar.ax.set_xlabel(label)
#label & title
plt.xlabel('Years after eruption')
plt.ylabel('Pressure (hPa)')
#additional formatting
ax = plt.gca()
ax.invert_yaxis()
plt.rc('font', **font)
plt.title(title)
#show/save plot
plt.rc({'figure.autolayout': True})
#show/save plot
ax.get_xaxis().majorTicks[-1].label1.set_horizontalalignment('right')
ax.get_yaxis().majorTicks[0].label1.set_verticalalignment('top')
plt.savefig(filename)
plt.close()
def latvtimea(data, cdata, variable, ncdf, controlncdf):
#anomaly
plt.figure(figsize=figsize)
data = calc.anomalize(data, cdata, ncdf, controlncdf)
#construct strings
label = labels[variable]
if variable in names:
title = names[variable]
else:
title = variable
subtitle = title
filename = (variable + '_' + filenames[ncdf.filename] + '_lvta_' +
filenames[controlncdf.filename] + '.eps')
#colormap
cmap = LinearSegmentedColormap('bwr', colors)
#calculate params
x, y = calc.getx(ncdf), calc.gety(ncdf)
z = data.transpose()
datamax, datamin = calc.getmax(data, True), calc.getmin(data, True)
v = np.linspace(datamin, datamax, 17)
#create plot
cp = plt.contourf(x, y, z, v, extend='both', colors=None, cmap=cmap,
vmin=datamin, vmax=datamax)
plt.xlim(x[0], x[-1])
plt.ylim(-90, 90)
#create colorbar
cbar = plt.colorbar(cp, orientation='horizontal',
aspect=15, format="%.2g", pad = .2)
cbar.ax.set_xlabel(label)
#label & title
plt.xlabel('Years after eruption')
plt.ylabel('Latitude')
#add control stats
'''cmean = stats.nanmean(cdata.flat)
#cmean = calc.anlmean(cdata)
cstd = stats.nanstd(cdata.flat)
#cstd = calc.anlstd(cdata)
string = ('Control mean: ' + '%.2g'%cmean +
'\nControl standard deviation: ' + '%.2g'%cstd)
plt.figtext(.02,.02,string, fontsize=12)'''
newlabels=[]
for label in cbar.ax.xaxis.get_ticklabels()[::2]:
newlabels.append(label.get_text())
cbar.ax.xaxis.set_ticklabels(newlabels)
newticks=[]
for tick in cbar.ax.get_xticks()[::2]:
newticks.append(tick)
cbar.ax.set_xticks(newticks)
#additional formating
ax = plt.gca()
ax.get_xaxis().majorTicks[-1].label1.set_horizontalalignment('right')
plt.yticks(np.linspace(-90, 90, 7, endpoint=True))
ax.get_yaxis().majorTicks[-1].label1.set_verticalalignment('top')
plt.rc('font', **font)
plt.title(subtitle)
#show/save plot
plt.savefig(filename, bbox_inches='tight')
plt.close()
