From 8a4811c75db79e8e4bf712a80f85ad03a8f7da7d Mon Sep 17 00:00:00 2001 From: Fabian Konstantinidis Date: Fri, 19 Jan 2024 10:22:12 +0100 Subject: [PATCH] manually added arrow visualization from PR https://github.com/KIT-MRT/behavior_generation_lecture_python/pull/16 --- .../utils/grid_plotting.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/src/behavior_generation_lecture_python/utils/grid_plotting.py b/src/behavior_generation_lecture_python/utils/grid_plotting.py index 023c378..2823275 100644 --- 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])