pyic_fig.py: Allow for atm pressure level data. Get vertical axes dimension from netcdf dimension.

tools/pyic_fig.py

@@ -139,8 +139,6 @@ import sys
 from pathlib import Path
 sys.path.append(f'{Path.home()}/pyicon/')
 import pyicon as pyic  
-from pathlib import Path
-from netCDF4 import Dataset
 print('Done modules.')
 
 arrange_axes = pyic.arrange_axes
@@ -240,6 +238,8 @@ elif 'lev' in data.dims:
   depth_name = 'lev'
 elif 'lev_2' in data.dims:
   depth_name = 'lev_2'
+elif 'plev' in data.dims:
+  depth_name = 'plev'
 else:
   depth_name = 'none'
 
@@ -325,7 +325,11 @@ if (iopts.title_right=='auto') and ('time' in ds[var].dims):
 if (iopts.title_center=='auto'):
   iopts.title_center = ''
 if (iopts.title_left=='auto') and (depth_name!='none'):
-  iopts.title_left = f'depth = {data[depth_name].data:.1f}m'
+  try:
+    depth_units = data[depth_name].units
+  except:
+    depth_units = ''
+  iopts.title_left = f'{depth_name} = {data[depth_name].data:.1f} {depth_units}'
 elif iopts.title_left=='auto':
   iopts.title_left = ''