manually added arrow visualization from PR KIT-MRT#16

fabikonsti · Jan 19, 2024 · 8a4811c · 8a4811c
1 parent db97165
commit 8a4811c
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/behavior_generation_lecture_python/utils/grid_plotting.py b/src/behavior_generation_lecture_python/utils/grid_plotting.py
@@ -70,7 +70,16 @@ def plot_grid_step(iteration):
                 scaling = np.sqrt(dx**2.0 + dy**2.0) * 2.5
                 dx /= scaling
                 dy /= scaling
-                plt.arrow(x, y, dx, dy)
+                plt.arrow(
+                    x,
+                    y,
+                    dx,
+                    dy,
+                    shape="full",
+                    lw=1.0,
+                    length_includes_head=True,
+                    head_width=0.15,
+                )
         plt.axis("equal")
         plt.xlim([0, columns])
         plt.ylim([0, rows])