Change index at DataFrame creation in DataCollector.get_agent_vars_da…

…taframe
projectmesa · Jan 21, 2023 · dcbc90b · dcbc90b
1 parent c7b2ccf
commit dcbc90b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mesa/datacollection.py b/mesa/datacollection.py
@@ -232,8 +232,8 @@ def get_agent_vars_dataframe(self):
         df = pd.DataFrame.from_records(
             data=all_records,
             columns=["Step", "AgentID"] + rep_names,
+            index=["Step", "AgentID"],
         )
-        df = df.set_index(["Step", "AgentID"])
         return df
 
     def get_table_dataframe(self, table_name):