Bugfix

plot_single_transmission_EM9190.py

@@ -28,6 +28,13 @@ tex_fonts = {
 #plt.rcParams.update(tex_fonts)
 
 
+def convert_cellid(value):
+    if isinstance(value, str):
+        return int(value.split(" ")[-1].replace("(", "").replace(")", ""))
+    else:
+        return int(-1)
+    
+
 if __name__ == "__main__":
     parser = ArgumentParser()
     parser.add_argument("-s", "--serial_file", required=True, help="Serial csv file.")
@@ -91,7 +98,7 @@ if __name__ == "__main__":
         #transmission_df = transmission_df.filter(["goodput", "datetime", "ack_rtt", "goodput_rolling", "snd_cwnd"])
 
         # read serial csv
-        serial_df = pd.read_csv(args.serial_file, dtype=dict(Cell_ID=str),)
+        serial_df = pd.read_csv(args.serial_file, converters={"Cell_ID": convert_cellid})
         serial_df["datetime"] = pd.to_datetime(serial_df["datetime"]) - pd.Timedelta(hours=1)
         serial_df = serial_df.set_index("datetime")
         serial_df.index = pd.to_datetime(serial_df.index)
@@ -99,8 +106,8 @@ if __name__ == "__main__":
 
         print(serial_df["Cell_ID"])
 
-        serial_df["Cell_ID"] = serial_df["Cell_ID"].apply(
-            lambda x: int(x.split(" ")[-1].replace("(", "").replace(")", "")))
+        #serial_df["Cell_ID"] = serial_df["Cell_ID"].apply(
+        #    lambda x: int(x.split(" ")[-1].replace("(", "").replace(")", "")))
 
         transmission_df = pd.merge_asof(
             transmission_df,