DebasishDhal99
commited on
Commit
•
fcf43ac
1
Parent(s):
5e24ac5
Update multi_agent_2D.py
Browse files- multi_agent_2D.py +3 -1
multi_agent_2D.py
CHANGED
@@ -124,9 +124,11 @@ def multi_agent_walk(agent_count, iters, step_size = 1, random_seed = None):
|
|
124 |
|
125 |
fig.canvas.draw()
|
126 |
image_array = np.array(fig.canvas.renderer.buffer_rgba())
|
|
|
|
|
127 |
|
128 |
try:
|
129 |
-
return image_array,
|
130 |
except:
|
131 |
return image_array, None
|
132 |
|
|
|
124 |
|
125 |
fig.canvas.draw()
|
126 |
image_array = np.array(fig.canvas.renderer.buffer_rgba())
|
127 |
+
csv_file = "2d_random_walk_coordinates.csv"
|
128 |
+
df.to_csv(csv_file, index=False)
|
129 |
|
130 |
try:
|
131 |
+
return image_array, csv_file
|
132 |
except:
|
133 |
return image_array, None
|
134 |
|