Merge pull request #44 from SWIFTSIM/show_additional_points_in_single_run_plots

Highlight additional points in single-run plots

Author: JBorrow

velociraptor/autoplotter/lines.py

@@ -327,11 +327,11 @@ def plot_line(
         )
 
         if self.scatter == "none" or errors is None:
-            ax.plot(centers, heights, label=label)
+            (line,) = ax.plot(centers, heights, label=label)
         elif self.scatter == "errorbar":
-            ax.errorbar(centers, heights, yerr=errors, label=label)
+            (line, *_) = ax.errorbar(centers, heights, yerr=errors, label=label)
         elif self.scatter == "errorbar_both":
-            ax.errorbar(
+            (line, *_) = ax.errorbar(
                 centers,
                 heights,
                 yerr=errors,
@@ -362,4 +362,10 @@ def plot_line(
                 linewidth=0.0,
             )
 
+        try:
+            ax.scatter(additional_x.value, additional_y.value, color=line.get_color())
+        # In case the line object is undefined
+        except NameError:
+            ax.scatter(additional_x.value, additional_y.value)
+
         return