improve output file

modules/util/util.py

@@ -563,6 +563,14 @@ def make_for_full_domain_predict(h5f, name_list=None, satellite='GOES16', domain
 # -------------------------------------------------------------------------------------------
 
 
+flt_level_ranges_str = {k: None for k in range(5)}
+flt_level_ranges_str[0] = '0_2000'
+flt_level_ranges_str[1] = '2000_4000'
+flt_level_ranges_str[2] = '4000_6000'
+flt_level_ranges_str[3] = '6000_8000'
+flt_level_ranges_str[4] = '8000_15000'
+
+
 def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lons, lats):
     outfile_name = output_dir + 'icing_prediction_'+clvrx_str_time+'.h5'
     h5f_out = h5py.File(outfile_name, 'w')
@@ -573,7 +581,7 @@ def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lon
     flt_lvls = list(preds_dct.keys())
     for flvl in flt_lvls:
         preds = preds_dct[flvl]
-        icing_pred_ds = h5f_out.create_dataset('icing_prediction_level_'+str(flvl), data=preds, dtype='i2')
+        icing_pred_ds = h5f_out.create_dataset('icing_prediction_level_'+flt_level_ranges_str[flvl], data=preds, dtype='i2')
         icing_pred_ds.attrs.create('coordinates', data='y x')
         icing_pred_ds.attrs.create('grid_mapping', data='Projection')
         icing_pred_ds.dims[0].label = dim_0_name
@@ -581,7 +589,7 @@ def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lon
 
     for flvl in flt_lvls:
         probs = probs_dct[flvl]
-        icing_prob_ds = h5f_out.create_dataset('icing_probability_level_'+str(flvl), data=probs, dtype='f4')
+        icing_prob_ds = h5f_out.create_dataset('icing_probability_level_'+flt_level_ranges_str[flvl], data=probs, dtype='f4')
         icing_prob_ds.attrs.create('coordinates', data='y x')
         icing_prob_ds.attrs.create('grid_mapping', data='Projection')
         icing_prob_ds.dims[0].label = dim_0_name