an.tsouchlos/kalman_filter
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added legend to Displayer

Browse Source
This commit is contained in:
Andreas Tsouchlos 2022-04-25 09:23:56 +02:00
parent f46730fdb3
commit 0c992ec1e0
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
display/dislpay_2d.py
View File

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