diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 9972615fb7d82b98e95a7941e52e2eb1227973f4..75dba9f2ae0e0deb6088a376e0402ca2d8e5feed 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -358,14 +358,19 @@ def analyze_all(ice_dct, no_ice_dct, neg_ice_dct): ice_times = [] ice_files_5_6 = [] ice_times_5_6 = [] + ice_keys_5_6 = [] ice_files_1 = [] ice_times_1 = [] + ice_keys_1 = [] ice_files_4 = [] ice_times_4 = [] + ice_keys_4 = [] ice_files_3 = [] ice_times_3 = [] + ice_keys_3 = [] ice_files_2 = [] ice_times_2 = [] + ice_keys_2 = [] for ts in list(ice_dct.keys()): try: ds = get_goes_datasource(ts) @@ -376,18 +381,23 @@ def analyze_all(ice_dct, no_ice_dct, neg_ice_dct): if tup[3] == 5 or tup[3] == 6: ice_files_5_6.append(goes_file) ice_times_5_6.append(t_0) + ice_keys_5_6.append(ts) elif tup[3] == 1: ice_files_1.append(goes_file) ice_times_1.append(t_0) + ice_keys_1.append(ts) elif tup[3] == 4: ice_files_4.append(goes_file) ice_times_4.append(t_0) + ice_keys_4.append(ts) elif tup[3] == 3: ice_files_3.append(goes_file) ice_times_3.append(t_0) + ice_keys_3.append(ts) else: ice_files_2.append(goes_file) ice_times_2.append(t_0) + ice_keys_2.append(ts) last_file = goes_file except Exception: continue @@ -449,6 +459,8 @@ def analyze_all(ice_dct, no_ice_dct, neg_ice_dct): no_ice_times = np.array(no_ice_times) neg_ice_times = np.array(neg_ice_times) + print('no ice: ', no_ice_times.shape) + print('neg ice: ', neg_ice_times.shape) itrsct_vals, comm1, comm2 = np.intersect1d(ice_times_1, ice_times_4, return_indices=True) print(itrsct_vals.shape)