add violin-plot

Author: c0rychu

examples/violin-plot.ipynb

@@ -16,8 +16,10 @@ build-backend = "uv_build"
 [dependency-groups]
 dev = [
     "black>=24.8.0",
+    "ipykernel>=6.29.5",
     "matplotlib>=3.7.5",
     "numpy>=1.24.4",
+    "pip>=25.0.1",
     "pre-commit>=3.5.0",
     "pytest>=8.3.5",
     "ruff>=0.14.0",

examples/violin-plot.png

@@ -0,0 +1,3 @@
+from .utils import Handles
+
+__all__ = [Handles]

pyproject.toml

@@ -0,0 +1,13 @@
+class Handles(list):
+    def __init__(self, *args):
+        super().__init__(*args)
+
+    def append_violinplot(self, violinplot, label):
+        self.append(self._create_patch_from_violinplot(violinplot, label))
+
+    def _create_patch_from_violinplot(self, violinplot, label):
+        from matplotlib.patches import Patch
+
+        body = violinplot["bodies"][0]
+        color = body.get_facecolor()[0]
+        return Patch(color=color, label=label)