fix: sort by value by as well if not none

OpenMS · Jan 6, 2025 · 9e69cd3 · 9e69cd3
1 parent ffb85bb
commit 9e69cd3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pyopenms_viz/_core.py b/pyopenms_viz/_core.py
@@ -584,8 +584,12 @@ def __init__(
             self.data[y] = self.data[y] / self.data[y].max() * 100
 
         self._check_and_aggregate_duplicates()
+
         # sort data by x so in order
-        self.data.sort_values(by=x, inplace=True)
+        if self.by is not None:
+            self.data.sort_values(by=[self.by, x], inplace=True)
+        else:
+            self.data.sort_values(by=x, inplace=True)
 
         self.plot(self.data, self.x, self.y, **kwargs)