| #plt.rcParams.update(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__": | if __name__ == "__main__": | ||||
| parser = ArgumentParser() | parser = ArgumentParser() | ||||
| parser.add_argument("-s", "--serial_file", required=True, help="Serial csv file.") | parser.add_argument("-s", "--serial_file", required=True, help="Serial csv file.") | ||||
| #transmission_df = transmission_df.filter(["goodput", "datetime", "ack_rtt", "goodput_rolling", "snd_cwnd"]) | #transmission_df = transmission_df.filter(["goodput", "datetime", "ack_rtt", "goodput_rolling", "snd_cwnd"]) | ||||
| # read serial csv | # 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["datetime"] = pd.to_datetime(serial_df["datetime"]) - pd.Timedelta(hours=1) | ||||
| serial_df = serial_df.set_index("datetime") | serial_df = serial_df.set_index("datetime") | ||||
| serial_df.index = pd.to_datetime(serial_df.index) | serial_df.index = pd.to_datetime(serial_df.index) | ||||
| print(serial_df["Cell_ID"]) | 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 = pd.merge_asof( | ||||
| transmission_df, | transmission_df, |