| 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 | from matplotlib import ticker |
|---|
| 9 | import matplotlib.colors as colors |
|---|
| 10 | from mpl_toolkits.basemap import Basemap, shiftgrid |
|---|
| 11 | from FV3_utils import * |
|---|
| 12 | from input import * |
|---|
| 13 | |
|---|
| 14 | fa='sans-serif' |
|---|
| 15 | hfont = {'fontname':'Arial'} |
|---|
| 16 | mpl.rc('font',family=fa) |
|---|
| 17 | mpl.rc('pdf',fonttype=42) |
|---|
| 18 | font=30 |
|---|
| 19 | cc=['k'] |
|---|
| 20 | pal=get_cmap(name="rainbow") |
|---|
| 21 | norm=colors.LogNorm() |
|---|
| 22 | #lvls=np.logspace(-6,-4,21) |
|---|
| 23 | norm=None #colors.LogNorm() |
|---|
| 24 | lvls=np.linspace(37,57,21) |
|---|
| 25 | |
|---|
| 26 | altitude=1 # in km |
|---|
| 27 | |
|---|
| 28 | ### Data |
|---|
| 29 | filename1=name+'_A.nc' |
|---|
| 30 | # name=name+'.nc' |
|---|
| 31 | filename2=name+'.nc' |
|---|
| 32 | print("Plot "+filename1) |
|---|
| 33 | nc1=Dataset(filename1) |
|---|
| 34 | nc2=Dataset(filename2) |
|---|
| 35 | ts=nc2.variables["temperature"][:,0,:,:] |
|---|
| 36 | # ts=nc2.variables["tsurf"][:,:,:] |
|---|
| 37 | u=nc1.variables["u"][:,:,:,:] |
|---|
| 38 | v=nc1.variables["v"][:,:,:,:] |
|---|
| 39 | lat=getvar(nc1,"latitude") |
|---|
| 40 | alt=getvar(nc1,"altitude") |
|---|
| 41 | lon=getvar(nc1,"longitude") |
|---|
| 42 | ps=nc2.variables["ps"][:,:] |
|---|
| 43 | # ps=nc2.variables["phisinit"][:,:]/0.6169/1000. # altitude km |
|---|
| 44 | |
|---|
| 45 | numalt=getind(altitude,alt) |
|---|
| 46 | print('numalt =',numalt,'altitude=',alt[numalt]) |
|---|
| 47 | u=u[:,numalt,:,:] |
|---|
| 48 | v=v[:,numalt,:,:] |
|---|
| 49 | u=np.mean(u,axis=0) |
|---|
| 50 | v=np.mean(v,axis=0) |
|---|
| 51 | ts=np.mean(ts,axis=0) |
|---|
| 52 | ps=np.mean(ps,axis=0) |
|---|
| 53 | |
|---|
| 54 | if lon[0]<0: |
|---|
| 55 | ts=switchlon(ts) |
|---|
| 56 | u=switchlon(u) |
|---|
| 57 | v=switchlon(v) |
|---|
| 58 | ps=switchlon(ps) |
|---|
| 59 | # topo=switchlon(topo) |
|---|
| 60 | lon=lon+180. |
|---|
| 61 | |
|---|
| 62 | ### Figure |
|---|
| 63 | fig=mpl.figure(figsize=(20, 10)) |
|---|
| 64 | # CF=mpl.contourf(lon, lat, ts,lvls,cmap=pal,norm=norm,extend='both') |
|---|
| 65 | # cbar=mpl.colorbar(CF, shrink=1, ticks=lvls[::2]) #,format="%1.0f") |
|---|
| 66 | # cbar.ax.set_title("[K]",y=1.04,fontsize=font) |
|---|
| 67 | # for t in cbar.ax.get_yticklabels(): |
|---|
| 68 | # t.set_fontsize(font) |
|---|
| 69 | # mpl.title('Surface temperatures and winds',fontsize=font) |
|---|
| 70 | |
|---|
| 71 | #vect=lvls |
|---|
| 72 | # CS=mpl.contour(lon,lat,topo,np.linspace(-6,6,21),colors='k',linewidths=0.5) |
|---|
| 73 | #lab=mpl.clabel(CS, inline=1, fontsize=20, fmt='%1.2e',inline_spacing=1) |
|---|
| 74 | #for l in lab: |
|---|
| 75 | # l.set_rotation(0) |
|---|
| 76 | |
|---|
| 77 | CF=mpl.contourf(lon, lat, ps,cmap=pal,norm=norm,extend='both') |
|---|
| 78 | cbar=mpl.colorbar(CF, label="Pa") |
|---|
| 79 | cbar.set_label("Pa", size=font) |
|---|
| 80 | for t in cbar.ax.get_yticklabels(): |
|---|
| 81 | t.set_fontsize(font) |
|---|
| 82 | mpl.title('Surface pressures and winds',fontsize=font) |
|---|
| 83 | |
|---|
| 84 | getwinds(lon,lat,u,v,1,1,100,0.002,5) |
|---|
| 85 | |
|---|
| 86 | mpl.grid() |
|---|
| 87 | mpl.ylabel(r'Latitude',labelpad=10,fontsize=font) |
|---|
| 88 | mpl.xlabel('Longitude',labelpad=10, fontsize=font) |
|---|
| 89 | pylab.ylim([-90,90]) |
|---|
| 90 | yticks=np.linspace(-90,90,13) |
|---|
| 91 | pylab.xlim([0,360]) |
|---|
| 92 | xticks=np.linspace(0,360,7) |
|---|
| 93 | mpl.yticks(yticks,fontsize=font) |
|---|
| 94 | mpl.xticks(xticks,fontsize=font) |
|---|
| 95 | mpl.tight_layout() |
|---|
| 96 | mpl.savefig('mapwinds1km.png',bbox_inches='tight',dpi=70) |
|---|
| 97 | mpl.show() |
|---|
| 98 | |
|---|
