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 |
|