diff --git a/code/utility.py b/code/utility.py index 68102d3..0f3ce19 100644 --- a/code/utility.py +++ b/code/utility.py @@ -23,3 +23,14 @@ def gaussian(x, mu, sig): y = np.exp(-np.power(x - mu, 2.) / (2 * np.power(sig, 2.))) return y +def map_keys(input): + df_map = {} + for k in input.keys(): + df = k[1] + ch = k[3] + if df in df_map.keys(): + df_map[df].append(k) + else: + df_map[df] = [k] + return df_map + print(ch)