Changes to inplace operations for large dataframes.

2023-01-11 09:03:41 +01:00
parent 17421f713e
commit c81e6635b4
1 changed files with 4 additions and 5 deletions
--- a/plot_transmission_timeline.py
+++ b/plot_transmission_timeline.py
@@ -194,11 +194,10 @@ if __name__ == "__main__":

    # Frequency
    stats_df = transmission_df \
-        .groupby("goodput") \
-        ["goodput"] \
-        .agg("count") \
-        .pipe(pd.DataFrame) \
-        .rename(columns={"goodput": 'frequency'})
+        .groupby("goodput", inplace=True)["goodput"] \
+        .agg("count", inplace=True) \
+        .pipe(pd.DataFrame, inplace=True) \
+        .rename(columns={"goodput": 'frequency'}, inplace=True)

    # PDF
    stats_df['pdf'] = stats_df['frequency'] / sum(stats_df['frequency'])