Keep order of columns for model and agent vars DataFrame

jligm-hash · Jan 24, 2021 · 2b8dbdb · 2b8dbdb
1 parent 8823e13
commit 2b8dbdb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/mesa/batchrunner.py b/mesa/batchrunner.py
@@ -257,10 +257,10 @@ def _prepare_report_table(self, vars_dict, extra_cols=None):
         column as a key.
         """
         extra_cols = ["Run"] + (extra_cols or [])
-        index_cols = set()
-        for params in self.parameters_list:
-            index_cols |= params.keys()
-        index_cols = list(index_cols) + extra_cols
+        index_cols = []
+        if self.parameters_list:
+            index_cols = list(self.parameters_list[0].keys())
+        index_cols += extra_cols
 
         records = []
         for param_key, values in vars_dict.items():