| 1 | #! /usr/bin/env python |
|---|
| 2 | from netCDF4 import Dataset |
|---|
| 3 | from numpy import * |
|---|
| 4 | import numpy as np |
|---|
| 5 | import matplotlib.pyplot as mpl |
|---|
| 6 | from matplotlib.cm import get_cmap |
|---|
| 7 | import pylab |
|---|
| 8 | import matplotlib.colors as mcolors |
|---|
| 9 | from FV3_utils import * |
|---|
| 10 | from input import * |
|---|
| 11 | |
|---|
| 12 | ############################ |
|---|
| 13 | var="ch4_gas" #variable |
|---|
| 14 | tint=[30,32] #Time must be as written in the input file |
|---|
| 15 | xarea="-180,179" |
|---|
| 16 | |
|---|
| 17 | nc1=Dataset(name+"_A.nc") |
|---|
| 18 | |
|---|
| 19 | lat=getvar(nc1,"latitude") |
|---|
| 20 | lon=getvar(nc1,"longitude") |
|---|
| 21 | alt=getvar(nc1,"altitude") |
|---|
| 22 | tim=getvar(nc1,"Time") |
|---|
| 23 | ############################ |
|---|
| 24 | |
|---|
| 25 | myvar=getvar(nc1,var,tint, l_mean=True, t_mean=True) |
|---|
| 26 | myvar=myvar*28/16.*100. |
|---|
| 27 | |
|---|
| 28 | mpl.figure(figsize=(20, 10)) |
|---|
| 29 | |
|---|
| 30 | font=26 |
|---|
| 31 | |
|---|
| 32 | #pal=rvb1 #get_cmap(name="RdYlBu_r") |
|---|
| 33 | #pal=get_cmap(name="Spectral_r") |
|---|
| 34 | pal=get_cmap(name="rainbow") |
|---|
| 35 | |
|---|
| 36 | xticks=[-90,-60,-30,0,30,60,90] |
|---|
| 37 | #yticks=np.linspace(0,240,9) |
|---|
| 38 | |
|---|
| 39 | mymin=0.1 |
|---|
| 40 | mymax=4 |
|---|
| 41 | |
|---|
| 42 | # log |
|---|
| 43 | norm=mcolors.LogNorm() |
|---|
| 44 | lvls=np.logspace(np.log10(mymin),np.log10(mymax),16) |
|---|
| 45 | #titi=[1.e-14,1.e-13,1.e-12,1.e-11] |
|---|
| 46 | CF=mpl.contourf(lat, alt, myvar,levels=lvls,norm=norm,cmap=pal) |
|---|
| 47 | cbar=mpl.colorbar(CF, shrink=1, format="%.1f",extend='both') |
|---|
| 48 | |
|---|
| 49 | for t in cbar.ax.get_yticklabels(): |
|---|
| 50 | t.set_fontsize(font) |
|---|
| 51 | |
|---|
| 52 | |
|---|
| 53 | vect=lvls |
|---|
| 54 | CS=mpl.contour(lat,alt,myvar,vect,colors='k',linewidths=0.5) |
|---|
| 55 | #### inline=1 : values over the line |
|---|
| 56 | mpl.clabel(CS, inline=1, fontsize=20, fmt='%1.1f',inline_spacing=1) |
|---|
| 57 | |
|---|
| 58 | |
|---|
| 59 | #lev=np.logspace(np.log10(0.1),np.log10(4),9) |
|---|
| 60 | |
|---|
| 61 | #mpl.title('Latitude ='+str(tintstr[i]),fontsize=font) |
|---|
| 62 | mpl.ylabel('Altitude (km)',labelpad=10,fontsize=font) |
|---|
| 63 | mpl.xlabel('Latitude (deg)',labelpad=10, fontsize=font) |
|---|
| 64 | mpl.xticks(xticks,fontsize=font) |
|---|
| 65 | #mpl.xticks(fontsize=font) |
|---|
| 66 | #mpl.yticks(yticks,fontsize=font) |
|---|
| 67 | mpl.yticks(fontsize=font) |
|---|
| 68 | pylab.ylim([-4,200]) |
|---|
| 69 | |
|---|
| 70 | mpl.savefig('meanvmrch4.eps',dpi=200) |
|---|
| 71 | mpl.savefig('meanvmrch4.png',dpi=200) |
|---|
| 72 | mpl.show() |
|---|
| 73 | |
|---|
| 74 | |
|---|
| 75 | |
|---|
| 76 | |
|---|