minor

modules/amv/aeolus.py

@@ -36,6 +36,7 @@ def compare_aeolus_max_height(aeolus_dict, files_path, grid_value_name='cld_heig
     num_files = len(clvrx_files.flist)
 
     keys = list(aeolus_dict.keys())
+    num_profs = len(keys)
 
     a_lons = [[] for i in range(num_files)]
     a_lats = [[] for i in range(num_files)]
@@ -62,15 +63,14 @@ def compare_aeolus_max_height(aeolus_dict, files_path, grid_value_name='cld_heig
 
         a_lons[f_idx].append(lon)
         a_lats[f_idx].append(lat)
-        a_profs[f_idx].append(layers)
         a_times[f_idx].append(key)
+        a_profs[f_idx].append(layers)
 
     for f_idx in range(num_files):
         a_lons[f_idx] = np.array(a_lons[f_idx])
         a_lats[f_idx] = np.array(a_lats[f_idx])
         a_hgts[f_idx] = np.array(a_hgts[f_idx])
         a_times[f_idx] = np.array(a_times[f_idx])
-        a_profs[f_idx] = np.array(a_profs[f_idx])
 
     grd_hgt = []
     grd_mean = []
@@ -114,9 +114,14 @@ def compare_aeolus_max_height(aeolus_dict, files_path, grid_value_name='cld_heig
         grd_crds = np.unravel_index(nn_idxs, grd_vals.shape)
         a_lons[f_idx] = a_lons[f_idx][on_earth]
         a_lats[f_idx] = a_lats[f_idx][on_earth]
-        a_profs[f_idx] = a_profs[f_idx][on_earth]
         a_times[f_idx] = a_times[f_idx][on_earth]
 
+        layers = []
+        for k in range(len(nn_idxs)):
+            if on_earth[k]:
+                layers.append(a_profs[f_idx][k])
+        a_profs[f_idx] = layers
+
         total += len(nn_idxs)
         # loop over Aeolus profile valid in the file
         for k in range(len(nn_idxs)):
@@ -210,7 +215,9 @@ def compare_aeolus_max_height(aeolus_dict, files_path, grid_value_name='cld_heig
             layer_top.append(tops)
 
         h5f.close()
+        print(f_idx, num_files)
 
+    print('num aeolus profiles: ', num_profs)
     print('Overall: ', num_hits, vld_grd, total)
     print('*')
     hits = np.array(hits)