From 97567140adc96eaaaf81874ddb266723bd5d2807 Mon Sep 17 00:00:00 2001 From: Lukas Prause Date: Thu, 16 Mar 2023 15:11:13 +0100 Subject: [PATCH] Plot pcid and scid. --- plot_single_transmission.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/plot_single_transmission.py b/plot_single_transmission.py index 3d67752..029eec4 100755 --- a/plot_single_transmission.py +++ b/plot_single_transmission.py @@ -122,6 +122,12 @@ if __name__ == "__main__": unique_cells = transmission_df["lte_cell_color"].unique() color_list = cmap.colors * (round(len(unique_cells) / len(cmap.colors)) + 1) + transmission_df["index"] = transmission_df.index + for c in transmission_df["lte_cell_color"].unique(): + bounds = transmission_df[["index", "lte_cell_color"]].groupby("lte_cell_color").agg(["min", "max"]).loc[ + c] + ax0.axvspan(bounds.min(), bounds.max(), alpha=0.1, color=color_list[c]) + # create list fo color indices for nr cells color_dict = dict() color_list = list() @@ -140,10 +146,6 @@ if __name__ == "__main__": unique_cells = transmission_df["nr_cell_color"].unique() color_list = cmap.colors * (round(len(unique_cells) / len(cmap.colors)) + 1) - transmission_df["index"] = transmission_df.index - for c in transmission_df["lte_cell_color"].unique(): - bounds = transmission_df[["index", "lte_cell_color"]].groupby("lte_cell_color").agg(["min", "max"]).loc[c] - ax0.axvspan(bounds.min(), bounds.max(), alpha=0.1, color=color_list[c]) for c in transmission_df["nr_cell_color"].unique(): bounds = transmission_df[["index", "nr_cell_color"]].groupby("nr_cell_color").agg(["min", "max"]).loc[c] ax00.axvspan(bounds.min(), bounds.max(), alpha=0.1, color=color_list[c])