From 0c992ec1e031bc633e551dffc9e7e665fc4c4e12 Mon Sep 17 00:00:00 2001 From: Andreas Tsouchlos Date: Mon, 25 Apr 2022 09:23:56 +0200 Subject: [PATCH] Added legend to Displayer --- display/dislpay_2d.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/display/dislpay_2d.py b/display/dislpay_2d.py index d6ef76e..747af87 100644 --- a/display/dislpay_2d.py +++ b/display/dislpay_2d.py @@ -37,7 +37,7 @@ class Displayer: def register_object(self, obj_name, obj_color='r'): - self._object_map[obj_name], = plt.plot([], [], color=obj_color, marker='o') + self._object_map[obj_name], = plt.plot([], [], color=obj_color, marker='o', label=obj_name) self._circle_map[obj_name] = plt.Circle((0, 0), 1, color=obj_color, fill=False) self._objects.append(self._object_map[obj_name]) @@ -61,6 +61,7 @@ class Displayer: def animate(self, n_steps, anim_callback): def anim_callback_wrapper(t): anim_callback(self, t) + self._ax.legend() return self._objects anim = FuncAnimation(self._fig,