diff --git a/modules/contrail/util.py b/modules/contrail/util.py index e26af9a8f2b786dbc2ba03c601c94b3392a56390..e1b09b2b66e1c11441dd25df08cf67f3705d65cf 100644 --- a/modules/contrail/util.py +++ b/modules/contrail/util.py @@ -62,8 +62,8 @@ def extract(mask_image, image_ts, clavrx_path): contrail_lats = contrail_lats[keep] # Indexes of contrail_press for individual bins - bins = np.arange(100, 1000, 50) - binned_indexes = np.digitize(contrail_press, bins) + press_bins = np.arange(100, 1000, 50) + binned_indexes = np.digitize(contrail_press, press_bins) # Store the indexes in a dictionary where the key is the bin number and value is the list of indexes bins_dict = {} @@ -111,7 +111,7 @@ def extract(mask_image, image_ts, clavrx_path): all_list = [] voxel_dict = {key: [] for key in bins_dict.keys()} for key in bins_dict.keys(): - press_level = bins[key] + press_level = press_bins[key] print('working on pressure level: ', press_level) for c_idx in bins_dict[key]: press = contrail_press[c_idx]