Add new ax feature

ioos · Jun 12, 2024 · c8d673a · c8d673a
1 parent 80f6759
commit c8d673a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/gliderpy/plotting.py b/gliderpy/plotting.py
@@ -46,6 +46,7 @@ def plot_track(df: pd.DataFrame) -> tuple(plt.Figure, plt.Axes):
 def plot_transect(
     df: pd.DataFrame,
     var: str,
+    ax: plt.Axes = None,
     **kw: dict,
 ) -> tuple(plt.Figure, plt.Axes):
     """Make a scatter plot of depth vs time coloured by a user defined
@@ -56,7 +57,11 @@ def plot_transect(
     """
     cmap = kw.get("cmap", None)
 
-    fig, ax = plt.subplots(figsize=(17, 2))
+    if ax is None:
+        fig, ax = plt.subplots(figsize=(17, 2))
+    else:
+        fig = ax.figure
+
     cs = ax.scatter(
         df.index,
         df["pressure"],