diff --git a/modules/amv/aeolus.py b/modules/amv/aeolus.py index 54ea8346a6808c3ddc344db36cc7ec3741983e4a..27f8b29c4602246a2ad1677385ddab7db83d842f 100644 --- a/modules/amv/aeolus.py +++ b/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)