diff --git a/modules/aeolus/aeolus_amv.py b/modules/aeolus/aeolus_amv.py index 959be3f89fe297e1439063f5ae9d7855a4be6789..6830069623361ceae17c315a370c147f183402ce 100644 --- a/modules/aeolus/aeolus_amv.py +++ b/modules/aeolus/aeolus_amv.py @@ -1062,12 +1062,6 @@ def compare_amvs_bestfit_all(all_list, bin_size=200): # amvs_list.append(tup[0]) # bfs_list.append(tup[1]) - amvs = np.concatenate(amvs_list) - bfs = np.concatenate(bfs_list) - print(amvs.shape, bfs.shape) - prd = np.concatenate(prd_list) - - print(amvs.shape, bfs.shape, prd.shape) # for tup in all_list: # amvs_list.append(tup[0]) @@ -1075,7 +1069,26 @@ def compare_amvs_bestfit_all(all_list, bin_size=200): # amvs = np.concatenate(amvs_list) # bfs = np.concatenate(bfs_list) + amvs = np.concatenate(amvs_list) + bfs = np.concatenate(bfs_list) + prd = np.concatenate(prd_list) + + thin = np.logical_and(prd[:, 2] > 0, prd[:, 2] < 1) + thick = np.logical_and(prd[:, 2] >= 1, prd[:, 2] < 18) + deep = np.logical_and(prd[:, 2] >= 18, prd[:, 2] <= 20) + + amvs_thin = amvs[thin, :] + amvs_thick = amvs[thick, :] + amvs_deep = amvs[deep, :] + + bfs_thin = bfs[thin, :] + bfs_thick = bfs[thick, :] + bfs_deep = bfs[deep, :] + bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs, bfs, bin_size=bin_size) + bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs_thin, bfs_thin, bin_size=bin_size) + bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs_thick, bfs_thick, bin_size=bin_size) + bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs_deep, bfs_deep, bin_size=bin_size) def compare_amvs_bestfit(amvs, bfs, bin_size=200):