chore: Proper axis labels for standard plots

iwishiwasaneagle · Apr 5, 2023 · 585d2d3 · 585d2d3
1 parent f6c03aa
commit 585d2d3
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/jdrones/plotting.py b/src/jdrones/plotting.py
@@ -266,16 +266,18 @@ def plot_standard(
         (Default = :code:`True`)
     """
     fig = plt.figure(figsize=figsize)
+
     ax = fig.add_subplot(221, projection="3d")
     plot_3d_path(df, ax)
 
-    for ind, states in (
-        (222, ("x", "y", "z")),
-        (223, ("vx", "vy", "vz")),
-        (224, ("phi", "theta", "psi")),
+    for ind, states, label in (
+        (222, ("x", "y", "z"), "position (m)"),
+        (223, ("vx", "vy", "vz"), "velocity (m/s)"),
+        (224, ("phi", "theta", "psi"), "angular position (rad)"),
     ):
         ax = fig.add_subplot(ind)
         plot_states_over_time(df, states, ax)
+        ax.set_ylabel(label)
         ax.legend()
 
     fig.tight_layout()