Adds stackplot for bandwith.

plot_single_transmission_EM9190.py

@@ -5,6 +5,7 @@ import os
 from argparse import ArgumentParser
 
 import matplotlib
+import numpy as np
 import pandas as pd
 import matplotlib.pyplot as plt
 
@@ -109,6 +110,13 @@ if __name__ == "__main__":
 
         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
         scaley = 1.5
         scalex = 1.0
@@ -152,6 +160,7 @@ if __name__ == "__main__":
         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")
         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")
 
         ax2.spines.right.set_position(("axes", 1.1))