From 6723350c8471fef9ecbf70ceca4e3f7e0d10e817 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Fri, 28 May 2021 13:47:06 -0500 Subject: [PATCH] minor --- modules/amv/aeolus.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/modules/amv/aeolus.py b/modules/amv/aeolus.py index 54ea8346..27f8b29c 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) -- GitLab