| from argparse import ArgumentParser | from argparse import ArgumentParser | ||||
| import matplotlib | import matplotlib | ||||
| import numpy as np | |||||
| import pandas as pd | import pandas as pd | ||||
| import matplotlib.pyplot as plt | import matplotlib.pyplot as plt | ||||
| transmission_df.index = transmission_df["arrival_time"] | transmission_df.index = transmission_df["arrival_time"] | ||||
| # replace 0 in RSRQ with Nan | |||||
| transmission_df = transmission_df["bw_sum"].replace(0, np.NaN) | |||||
| # stacked plot for bandwidth | |||||
| transmission_df["lte_bw_sum"] = transmission_df["bw_sum"] - transmission_df["NR5G_dl_bw"] | |||||
| transmission_df["nr_bw_sum"] = transmission_df["NR5G_dl_bw"] | |||||
| # transmission timeline | # transmission timeline | ||||
| scaley = 1.5 | scaley = 1.5 | ||||
| scalex = 1.0 | scalex = 1.0 | ||||
| ax2.plot(transmission_df["goodput_rolling"], color="blue", linestyle="solid", label="goodput") | ax2.plot(transmission_df["goodput_rolling"], color="blue", linestyle="solid", label="goodput") | ||||
| ax00.plot(transmission_df["NR5G_RSRQ_(dB)"].dropna(), color="magenta", linestyle="dotted", label="NR RSRQ") | ax00.plot(transmission_df["NR5G_RSRQ_(dB)"].dropna(), color="magenta", linestyle="dotted", label="NR RSRQ") | ||||
| ax01.plot(transmission_df["bw_sum"].dropna(), color="peru", linestyle="solid", label="bandwidth") | ax01.plot(transmission_df["bw_sum"].dropna(), color="peru", linestyle="solid", label="bandwidth") | ||||
| ax01.stackplot(transmission_df["lte_bw_sum"], transmission_df["nr_bw_sum"], colors=["lightsteelblue", "cornflowerblue"], labels=["4G", "5G"]) | |||||
| ax02.plot(transmission_df["RSRQ_(dB)"].dropna(), color="purple", linestyle="dotted", label="LTE RSRQ") | ax02.plot(transmission_df["RSRQ_(dB)"].dropna(), color="purple", linestyle="dotted", label="LTE RSRQ") | ||||
| ax2.spines.right.set_position(("axes", 1.1)) | ax2.spines.right.set_position(("axes", 1.1)) |