diff --git a/modules/amv/intercompare.py b/modules/amv/intercompare.py index 608c0f2bea8dfc2844c273c80f723a22ee801b92..eaf15ead1a9cdfe2c219b770152098f76981f2bd 100644 --- a/modules/amv/intercompare.py +++ b/modules/amv/intercompare.py @@ -251,7 +251,7 @@ km_per_deg = 111.0 # Find raobs within threshold distance for each (lat, lon) point -def get_raob_neighbors(lats, lons, raob_dct): +def get_raob_neighbors(lats, lons, raob_dct, dist_threshold=150): nbor_dct = {} num_pts = lats.shape[0] @@ -410,11 +410,11 @@ def amv_raob_diff(amv_lat, amv_lon, amv_spd, amv_dir, amv_press, raob_dct): return spd_diffs, dir_diffs -def run_best_fit(amvs, raob_dct, min_num_levs=50): +def run_best_fit(amvs, raob_dct, dist_threshold=200, min_num_levs=20): num_amvs = amvs.shape[0] bestfits = [] - raob_nbor_dct = get_raob_neighbors(amvs[:, amv_lat_idx], amvs[:, amv_lon_idx], raob_dct) + raob_nbor_dct = get_raob_neighbors(amvs[:, amv_lat_idx], amvs[:, amv_lon_idx], raob_dct, dist_threshold=dist_threshold) for j in range(num_amvs): tup = raob_nbor_dct[j]