From 1b0c0b9c63ba804cbd6cebf58cff75738ef74860 Mon Sep 17 00:00:00 2001 From: Langspielplatte Date: Tue, 28 Feb 2023 09:03:18 +0100 Subject: [PATCH] Bugfixes --- cdf_compare.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/cdf_compare.py b/cdf_compare.py index e8c17e4..6fc9b35 100755 --- a/cdf_compare.py +++ b/cdf_compare.py @@ -46,7 +46,7 @@ def chunk(it, size): return iter(lambda: tuple(islice(it, size)), ()) -def plot_cdf(dataframe, column_name): +def plot_cdf(dataframe, column_name, axis=None): stats_df = dataframe \ .groupby(column_name) \ [column_name] \ @@ -61,7 +61,11 @@ def plot_cdf(dataframe, column_name): stats_df["CDF"] = stats_df["PDF"].cumsum() stats_df = stats_df.reset_index() - stats_df.plot(x=column_name, y=["CDF"], grid=True) + if axis: + stats_df.plot(x=column_name, y=["CDF"], grid=True, ax=axis) + else: + stats_df.plot(x=column_name, y=["CDF"], grid=True) + return axis if __name__ == "__main__": @@ -186,8 +190,8 @@ if __name__ == "__main__": )) # Plot sRTT CDF - plot_cdf(transmission_df_list[0]["df"], "srtt") - plot_cdf(transmission_df_list[1]["df"], "srtt") + ax = plot_cdf(transmission_df_list[0]["df"], "srtt") + plot_cdf(transmission_df_list[1]["df"], "srtt", axis=ax) plt.xscale("log") plt.xlabel("sRTT [s]") plt.ylabel("CDF") @@ -198,8 +202,8 @@ if __name__ == "__main__": plt.clf() # Plot goodput CDF - plot_cdf(transmission_df_list[0]["df"], "goodput") - plot_cdf(transmission_df_list[1]["df"], "goodput") + ax = plot_cdf(transmission_df_list[0]["df"], "goodput") + plot_cdf(transmission_df_list[1]["df"], "goodput", axis=ax) plt.xlabel("goodput [mbps]") plt.ylabel("CDF") plt.legend([transmission_df_list[0]["cc_algo"], transmission_df_list[1]["cc_algo"]])