diff --git a/modules/aeolus/aeolus_amv.py b/modules/aeolus/aeolus_amv.py index df7832e6728274e31f00d50af319556f1f97b26b..0062564c12c2dd790c5fac00190e0e306a8d6480 100644 --- a/modules/aeolus/aeolus_amv.py +++ b/modules/aeolus/aeolus_amv.py @@ -469,9 +469,9 @@ def run_best_fit_all(): for k, file in enumerate(raob_files): raob_dct, ts = get_raob_dict_cdf(raob_dir+file) m_d = match_amvs_to_raobs(raob_dct, ts, amv_files=amv_files) - dct = run_best_fit(m_d, raob_dct, gfs_dir+gfs_files[k]) - out_list.append(dct) + bf_dct = run_best_fit(m_d, raob_dct, gfs_dir+gfs_files[k]) prd_dct = get_product_at_locs(m_d, ts, prd_files) + out_list.append((bf_dct, prd_dct)) amvs = get_amvs(amv_files, ts) bfs = run_best_fit_gfs(amvs, gfs_dir+gfs_files[k], amv_lat_idx=0, amv_lon_idx=1, amv_prs_idx=4, amv_spd_idx=5, amv_dir_idx=6) @@ -1047,6 +1047,12 @@ def compare_amvs_bestfit_all(all_list, bin_size=200): amvs = np.concatenate(amvs_list) bfs = np.concatenate(bfs_list) + # for tup in all_list: + # amvs_list.append(tup[0]) + # bfs_list.append(tup[1]) + # amvs = np.concatenate(amvs_list) + # bfs = np.concatenate(bfs_list) + bin_ranges, bin_pres, bin_spd, bin_dir = compare_amvs_bestfit(amvs, bfs, bin_size=bin_size)