fix(matplotlib): use inline data instead of seaborn for scatter-color-groups

Author: MarkusNeusinger

plots/matplotlib/scatter/scatter-color-groups/default.py

@@ -4,11 +4,27 @@
 """
 
 import matplotlib.pyplot as plt
-import seaborn as sns
-
-
-# Data
-data = sns.load_dataset("iris")
+import numpy as np
+import pandas as pd
+
+
+# Data - Iris-like dataset
+np.random.seed(42)
+n_per_group = 50
+
+data = pd.DataFrame({
+    "sepal_length": np.concatenate([
+        np.random.normal(5.0, 0.35, n_per_group),
+        np.random.normal(5.9, 0.50, n_per_group),
+        np.random.normal(6.6, 0.60, n_per_group),
+    ]),
+    "sepal_width": np.concatenate([
+        np.random.normal(3.4, 0.38, n_per_group),
+        np.random.normal(2.8, 0.30, n_per_group),
+        np.random.normal(3.0, 0.30, n_per_group),
+    ]),
+    "species": ["setosa"] * n_per_group + ["versicolor"] * n_per_group + ["virginica"] * n_per_group,
+})
 
 # Color palette (colorblind safe from style guide)
 colors = ["#306998", "#FFD43B", "#DC2626"]