diff --git a/modules/util/util.py b/modules/util/util.py index 49f435aeda8d874753ab2b118d7d40479d587901..298ef6eda0b71a7e24f3608db9775de0aca52385 100644 --- a/modules/util/util.py +++ b/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