[7715] | 1 | from netCDF4 import Dataset |
---|
| 2 | import numpy as np |
---|
| 3 | from numpy import ma |
---|
| 4 | import argparse |
---|
| 5 | import matplotlib.pyplot as plt |
---|
| 6 | import matplotlib |
---|
| 7 | |
---|
| 8 | parser = argparse.ArgumentParser() |
---|
| 9 | parser.add_argument("-f" , metavar='file_name' , help="names of input files" , type=str , nargs="+", required=True ) |
---|
| 10 | parser.add_argument("-v" , metavar='var_name' , help="variable list" , type=str , nargs=1 , required=True ) |
---|
| 11 | args = parser.parse_args() |
---|
| 12 | |
---|
| 13 | # read mesh_mask |
---|
| 14 | ncid = Dataset('mesh_mask.nc') |
---|
| 15 | lat2d = ncid.variables['gphit' ][ :,:].squeeze() |
---|
| 16 | lon2d = ncid.variables['glamt' ][ :,:].squeeze() |
---|
| 17 | msk = ncid.variables['tmaskutil'][0,:,:].squeeze() |
---|
| 18 | ncid.close() |
---|
| 19 | |
---|
| 20 | plt.figure(figsize=np.array([210,210]) / 25.4) |
---|
| 21 | |
---|
| 22 | # read psi.nc |
---|
| 23 | ncid = Dataset(args.f[0]) |
---|
| 24 | var2d = ncid.variables[args.v[0]][-1,:,:].squeeze() |
---|
| 25 | var2dm = ma.masked_where(msk==0.0,var2d) |
---|
| 26 | # convert in m/y |
---|
| 27 | var2dm = var2dm * 86400 * 365 / 1e3 |
---|
| 28 | ncid.close() |
---|
| 29 | |
---|
| 30 | # define colorbar |
---|
| 31 | vlevel=np.arange(-1.6,1.8,0.2) |
---|
| 32 | pcol = plt.contourf(lon2d,lat2d,var2dm,levels=vlevel,extend='both') |
---|
| 33 | vlevel=np.arange(-1.6,1.8,0.4) |
---|
| 34 | matplotlib.rcParams['contour.negative_linestyle'] = 'solid' |
---|
| 35 | plt.contour(lon2d,lat2d,var2dm,levels=vlevel,colors='k') |
---|
| 36 | plt.grid() |
---|
| 37 | plt.title('melt rate ISOMIP (m/y)') |
---|
| 38 | plt.ylabel('Latitude',fontsize=14) |
---|
| 39 | plt.xlabel('Longitude',fontsize=14) |
---|
| 40 | cbar = plt.colorbar(pcol, ticks=vlevel) |
---|
| 41 | cbar.ax.tick_params(labelsize=14) |
---|
| 42 | |
---|
| 43 | # save figure |
---|
| 44 | plt.savefig('mlt.png', format='png', dpi=300) |
---|
| 45 | |
---|
| 46 | plt.show() |
---|
| 47 | |
---|