Modifies QuickPlot.plot to take a list of times and show superimposed plots in case of 1D variables.

CHANGELOG.md

@@ -14,6 +14,7 @@
 - Added phase-dependent particle options to LAM ([#4369](https://github.com/pybamm-team/PyBaMM/pull/4369))
 - Added a lithium ion equivalent circuit model with split open circuit voltages for each electrode (`SplitOCVR`). ([#4330](https://github.com/pybamm-team/PyBaMM/pull/4330))
 - Added the `pybamm.DiscreteTimeSum` expression node to sum an expression over a sequence of data times, and accompanying `pybamm.DiscreteTimeData` class to store the data times and values ([#4501](https://github.com/pybamm-team/PyBaMM/pull/4501))
+- Modified `quick_plot.plot` to accept a list of times and generate superimposed graphs for specified time points.([#4529](https://github.com/pybamm-team/PyBaMM/pull/4529))
 
 ## Optimizations
 

src/pybamm/plotting/quick_plot.py

@@ -479,13 +479,13 @@ def reset_axis(self):
             ):  # pragma: no cover
                 raise ValueError(f"Axis limits cannot be NaN for variables '{key}'")
 
-    def plot(self, t, dynamic=False):
+    def plot(self, t: float | list[float], dynamic: bool = False):
         """Produces a quick plot with the internal states at time t.
 
         Parameters
         ----------
-        t : float or list
-            Dimensional time (in 'time_units') at which to plot. Can be a single time or a list of times.
+        t : float or list of float
+        Dimensional time (in 'time_units') at which to plot. Can be a single time or a list of times.
         dynamic : bool, optional
             Determine whether to allocate space for a slider at the bottom of the plot when generating a dynamic plot.
             If True, creates a dynamic plot with a slider.
@@ -494,7 +494,7 @@ def plot(self, t, dynamic=False):
         plt = import_optional_dependency("matplotlib.pyplot")
         gridspec = import_optional_dependency("matplotlib.gridspec")
 
-        if isinstance(t, (int, float)):
+        if not isinstance(t, list):
             t = [t]
 
         self.fig = plt.figure(figsize=self.figsize)
@@ -597,7 +597,7 @@ def plot(self, t, dynamic=False):
                 spatial_vars = self.spatial_variable_dict[key]
                 variable = variable_lists[0][0]
 
-                for _idx, t_single in enumerate(t):
+                for t_single in t:
                     t_in_seconds = t_single * self.time_scaling_factor
                     x = self.first_spatial_variable[key]
                     y = self.second_spatial_variable[key]