| 1 | #! /usr/bin/env python |
|---|
| 2 | from netCDF4 import Dataset |
|---|
| 3 | from numpy import * |
|---|
| 4 | import numpy as np |
|---|
| 5 | import matplotlib |
|---|
| 6 | import matplotlib.pyplot as mpl |
|---|
| 7 | from matplotlib.cm import get_cmap |
|---|
| 8 | import pylab |
|---|
| 9 | from matplotlib import ticker |
|---|
| 10 | import matplotlib.colors as colors |
|---|
| 11 | import datetime |
|---|
| 12 | from mpl_toolkits.basemap import Basemap, shiftgrid |
|---|
| 13 | from FV3_utils import * |
|---|
| 14 | from input import * |
|---|
| 15 | |
|---|
| 16 | ############################ |
|---|
| 17 | folder="../" |
|---|
| 18 | name=name+'_A.nc' |
|---|
| 19 | # filename1=folder+"diagfi_mean_A.nc" |
|---|
| 20 | var="temperature" #variable |
|---|
| 21 | tint=[30,35] #Time must be as written in the input file |
|---|
| 22 | xarea="-180,179" |
|---|
| 23 | print(name) |
|---|
| 24 | nc1=Dataset(name) |
|---|
| 25 | |
|---|
| 26 | # max altitude in plot (maxed to max of alt) |
|---|
| 27 | max_alt = 230 |
|---|
| 28 | |
|---|
| 29 | |
|---|
| 30 | lat=getvar(nc1,"latitude") |
|---|
| 31 | lon=getvar(nc1,"longitude") |
|---|
| 32 | alt=getvar(nc1,"altitude") |
|---|
| 33 | tim=getvar(nc1,"Time") |
|---|
| 34 | ############################ |
|---|
| 35 | |
|---|
| 36 | mpl.figure(figsize=(20, 10)) |
|---|
| 37 | |
|---|
| 38 | myvar=getvar(nc1,var,tint,l_mean=True,t_mean=True) |
|---|
| 39 | font=26 |
|---|
| 40 | |
|---|
| 41 | pal=get_cmap(name="RdYlBu_r") |
|---|
| 42 | # pal=get_cmap(name="cividis") |
|---|
| 43 | # pal=get_cmap(name="hot") |
|---|
| 44 | # pal=get_cmap(name="gnuplot2") |
|---|
| 45 | lev=np.linspace(40,110,11) |
|---|
| 46 | newlon=lon+180 |
|---|
| 47 | xticks=[-90,-60,-30,0,30,60,90] |
|---|
| 48 | #yticks=np.linspace(0,240,9) |
|---|
| 49 | # alt=alt/1000. |
|---|
| 50 | |
|---|
| 51 | CF=mpl.contourf(lat,alt,myvar,lev,cmap=pal,extend='both') |
|---|
| 52 | cbar=mpl.colorbar(CF,shrink=1, format="%1.0f") |
|---|
| 53 | cbar.ax.set_title("T [K]",y=1.04,fontsize=font,pad=20) |
|---|
| 54 | for t in cbar.ax.get_yticklabels(): |
|---|
| 55 | t.set_fontsize(font) |
|---|
| 56 | |
|---|
| 57 | vect=lev |
|---|
| 58 | CS=mpl.contour(lat,alt,myvar,vect,colors='k',linewidths=0.5) |
|---|
| 59 | #inline=1 : values over the line |
|---|
| 60 | #mpl.clabel(CS, inline=2, fontsize=10, fmt='%1.1e') |
|---|
| 61 | mpl.clabel(CS, inline=1, fontsize=20, fmt='%1.0f',inline_spacing=1) |
|---|
| 62 | |
|---|
| 63 | mpl.title('Zonal mean temperature',fontsize=font, pad=20) |
|---|
| 64 | #mpl.title('Latitude ='+str(tintstr[i]),fontsize=font) |
|---|
| 65 | mpl.ylabel('Altitude (km)',labelpad=10,fontsize=font) |
|---|
| 66 | mpl.xlabel('Latitude (deg)',labelpad=10, fontsize=font) |
|---|
| 67 | mpl.xticks(xticks,fontsize=font) |
|---|
| 68 | #mpl.yticks(yticks,fontsize=font) |
|---|
| 69 | mpl.yticks(fontsize=font) |
|---|
| 70 | if type(ma.getmask(myvar)) in (bool,np.bool_): |
|---|
| 71 | max_alt = min(max_alt,max(alt)) |
|---|
| 72 | else: |
|---|
| 73 | max_alt = min(max_alt,max(alt[~ma.getmask(myvar)[:,0]][:-1])) |
|---|
| 74 | pylab.ylim([-4,max_alt]) |
|---|
| 75 | |
|---|
| 76 | mpl.savefig('tempsec.eps',dpi=200) |
|---|
| 77 | mpl.savefig('tempsec.png',dpi=200) |
|---|
| 78 | # mpl.show() |
|---|
| 79 | |
|---|
| 80 | |
|---|
| 81 | |
|---|
| 82 | |
|---|
