diff --git a/modules/contrail/util.py b/modules/contrail/util.py index 368b7f17f04cc8ccc27808db436a0066a7a801c3..20233e5aaf9567dbe3ff6e8a444f90b59bf04f29 100644 --- a/modules/contrail/util.py +++ b/modules/contrail/util.py @@ -69,7 +69,7 @@ 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, 100) + bins = np.arange(100, 1000, 50) binned_indexes = np.digitize(contrail_press, bins) # Store the indexes in a dictionary where the key is the bin number and value is the list of indexes @@ -110,6 +110,14 @@ def extract(mask_image, image_ts, clavrx_path): horz_wind_spd_3d = wind_speed(uwind_3d, vwind_3d) vert_shear_3d = first_derivative(horz_wind_spd_3d, axis=0, x=temp_3d.coords['Pressure'] * units.hPa) + voxel_dict = {key: [] for key in bins_dict.keys()} + for key in bins_dict.keys(): + print('working on pressure level: ', bins[key]) + for c_idx in bins_dict[key]: + lon = contrail_lons[c_idx] + lat = contrail_lats[c_idx] + press = contrail_press[c_idx] + xr_dataset.close()