add H2OLNZ in zonal mean 2d stratosphere (#868)

* add H2OLNZ

* refine contour levels

* fix UnboundLocalError in testing

Author: chengzhuzhang

e3sm_diags/driver/default_diags/zonal_mean_2d_stratosphere_model_vs_obs.cfg

@@ -40,6 +40,40 @@ seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "1
 contour_levels = [0.6,0.8, 1, 1.2, 1.6, 2, 2.5, 3, 3.5, 4]
 diff_levels = [-3, -2,-1.5,-1,-0.5,-0.1,0.1,0.5,1,1.5,2,3]
 
+[#]
+sets = ["zonal_mean_2d_stratosphere"]
+case_id = "MERRA2"
+variables = ["H2OLNZ"]
+ref_name = "MERRA2"
+reference_name = "MERRA2 Reanalysis"
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+contour_levels = [0.001, 0.0015, 0.002,0.0025,0.003,0.0035,0.004,0.0045,0.005]
+diff_levels = [-0.001, -0.00075, -0.00050, -0.00025, -0.00005,0.00005,0.00025, 0.0005,0.00075,0.001]
+
+[#]
+sets = ["zonal_mean_2d_stratosphere"]
+case_id = "MERRA2 (relative difference)"
+variables = ["Q"]
+ref_name = "MERRA2"
+reference_name = "MERRA2 Reanalysis"
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+diff_type = "relative"
+diff_title = "(Model - Obs.)/Obs. * 100"
+contour_levels = [0.6,0.8, 1, 1.2, 1.6, 2, 2.5, 3, 3.5, 4]
+diff_levels = [-100, -75, -50, -30, -15, -5, 5, 15, 30, 50, 75, 100]
+
+[#]
+sets = ["zonal_mean_2d_stratosphere"]
+case_id = "MERRA2 (relative difference)"
+variables = ["H2OLNZ"]
+ref_name = "MERRA2"
+reference_name = "MERRA2 Reanalysis"
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+diff_type = "relative"
+diff_title = "(Model - Obs.)/Obs. * 100"
+contour_levels = [0.001, 0.0015, 0.002,0.0025,0.003,0.0035,0.004,0.0045,0.005]
+diff_levels = [-100, -75, -50, -30, -15, -5, 5, 15, 30, 50, 75, 100]
+
 [#]
 sets = ["zonal_mean_2d_stratosphere"]
 case_id = "MERRA2"
@@ -105,3 +139,37 @@ reference_name = "ERA5 Reanalysis"
 seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
 contour_levels = [0.6,0.8, 1, 1.2, 1.6, 2, 2.5, 3, 3.5, 4]
 diff_levels = [-3, -2,-1.5,-1,-0.5,-0.1,0.1,0.5,1,1.5,2,3]
+
+[#]
+sets = ["zonal_mean_2d_stratosphere"]
+case_id = "ERA5"
+variables = ["H2OLNZ"]
+ref_name = "ERA5"
+reference_name = "ERA5 Reanalysis"
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+contour_levels = [0.001, 0.0015, 0.002,0.0025,0.003,0.0035,0.004,0.0045,0.005]
+diff_levels = [-0.001, -0.00075, -0.00050, -0.00025, -0.00005,0.00005,0.00025, 0.0005,0.00075,0.001]
+
+[#]
+sets = ["zonal_mean_2d_stratosphere"]
+case_id = "ERA5 (relative difference)"
+variables = ["Q"]
+ref_name = "ERA5"
+reference_name = "ERA5 Reanalysis"
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+diff_type = "relative"
+diff_title = "(Model - Obs.)/Obs. * 100"
+contour_levels = [0.6,0.8, 1, 1.2, 1.6, 2, 2.5, 3, 3.5, 4]
+diff_levels = [-100, -75, -50, -30, -15, -5, 5, 15, 30, 50, 75, 100]
+
+[#]
+sets = ["zonal_mean_2d_stratosphere"]
+case_id = "ERA5 (relative difference)"
+variables = ["H2OLNZ"]
+ref_name = "ERA5"
+reference_name = "ERA5 Reanalysis"
+seasons = ["ANN", "01", "02", "03", "04", "05", "06", "07", "08", "09", "10", "11", "12", "DJF", "MAM", "JJA", "SON"]
+diff_type = "relative"
+diff_title = "(Model - Obs.)/Obs. * 100"
+contour_levels = [0.001, 0.0015, 0.002,0.0025,0.003,0.0035,0.004,0.0045,0.005]
+diff_levels = [-100, -75, -50, -30, -15, -5, 5, 15, 30, 50, 75, 100]

e3sm_diags/plot/cartopy/arm_diags_plot.py

@@ -149,7 +149,7 @@ def plot_convection_onset_statistics(
         cwv_min = 28
         bin_width = 2.0
         sitename = "Darwin"
-    if region == "sgp":  # sgp
+    if region == "sgpc1":  # sgp
         cwv_max = 75
         cwv_min = 20
         bin_width = 2.0