fix replicates sorting

PEtab-dev · Dec 11, 2023 · 3b9d89c · 3b9d89c
1 parent c0158ba
commit 3b9d89c
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/petab/visualize/plotter.py b/petab/visualize/plotter.py
@@ -123,12 +123,22 @@ def generate_lineplot(
                 replicates = np.stack(
                     measurements_to_plot.data_to_plot.repl.values
                 )
+                cond, replicates = zip(
+                    *sorted(
+                        zip(
+                            measurements_to_plot.conditions,
+                            replicates
+                        )
+                    )
+                )
+                replicates = np.stack(replicates)
+
                 if replicates.ndim == 1:
                     replicates = np.expand_dims(replicates, axis=1)
 
                 # plot first replicate
                 p = ax.plot(
-                    measurements_to_plot.conditions,
+                    cond,
                     replicates[:, 0],
                     linestyle="-.",
                     marker="x",
@@ -138,7 +148,7 @@ def generate_lineplot(
 
                 # plot other replicates with the same color
                 ax.plot(
-                    measurements_to_plot.conditions,
+                    cond,
                     replicates[:, 1:],
                     linestyle="-.",
                     marker="x",