snapshot...

modules/aeolus/aeolus_amv.py

@@ -2118,6 +2118,8 @@ def create_bestfit_file(filename, match_dct, raob_dct, raob_ts, gfs_at_raob_dct,
     dist_to_raob = rootgrp.createVariable('dist_to_raob', 'f4', ['time', 'num_raob_profs', 'max_num_amvs'])
     dist_to_raob.units = 'km'
 
+    amv_source_index = rootgrp.createVariable('amv_source_index', 'f4', ['time', 'num_raob_profs', 'max_num_amvs'])
+
     var_s = rg_exmpl.variables
     for pidx, param in enumerate(amv_params):
         v = var_s[param]
@@ -2225,6 +2227,12 @@ def create_bestfit_file(filename, match_dct, raob_dct, raob_ts, gfs_at_raob_dct,
 
         dist_to_raob[0, idx, 0:num_amvs] = haversine_np(key.lon, key.lat, amvs[:, 0], amvs[:, 1])
 
+        param_s = match_dct.get(key)
+        idx_idx = param_s.shape[0] - 1
+        amv_idxs = param_s[idx_idx, ]
+
+        amv_source_index[0, idx, 0:num_amvs] = amv_idxs[:]
+
     rg_exmpl.close()
     rootgrp.close()