From 4bd1f5f1598cb316adfc09f630bfe0eaabc2ab37 Mon Sep 17 00:00:00 2001 From: Youngjoon Lee <5462944+youngjoon-lee@users.noreply.github.com> Date: Wed, 14 Aug 2024 11:29:04 +0900 Subject: [PATCH] fix: queuesim_stats: filter out topology csv files --- mixnet/queuesim/statistics.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/mixnet/queuesim/statistics.py b/mixnet/queuesim/statistics.py index 2601810..56f9bcb 100644 --- a/mixnet/queuesim/statistics.py +++ b/mixnet/queuesim/statistics.py @@ -1,5 +1,6 @@ import glob import os +import re import pandas as pd @@ -43,8 +44,10 @@ def __calculate_paramset_stats(paramset_dir: str, session_result_path: str): info = pd.read_csv(f"{paramset_dir}/paramset.csv") series_list = [] - for iter_csv in glob.glob(f"{paramset_dir}/iteration_*.csv"): - df = pd.read_csv(iter_csv) + for iter_csv in [ + f for f in os.listdir(paramset_dir) if re.match(r"iteration_\d+.csv", f) + ]: + df = pd.read_csv(f"{paramset_dir}/{iter_csv}") # The 1st column is the dissemination time series_list.append(pd.Series(df.iloc[:, 0]))