DOC make plot_mean_shift.py more colourblind friendly (scikit-learn#24553)

rprkh · glemaitre · commit 389cfd5fbbf0 · 2022-10-31T11:25:37.000+01:00
diff --git a/examples/cluster/plot_mean_shift.py b/examples/cluster/plot_mean_shift.py
@@ -42,20 +42,21 @@
 # Plot result
 # -----------
 import matplotlib.pyplot as plt
-from itertools import cycle
 
 plt.figure(1)
 plt.clf()
 
-colors = cycle("bgrcmyk")
+colors = ["#dede00", "#377eb8", "#f781bf"]
+markers = ["x", "o", "^"]
+
 for k, col in zip(range(n_clusters_), colors):
     my_members = labels == k
     cluster_center = cluster_centers[k]
-    plt.plot(X[my_members, 0], X[my_members, 1], col + ".")
+    plt.plot(X[my_members, 0], X[my_members, 1], markers[k], color=col)
     plt.plot(
         cluster_center[0],
         cluster_center[1],
-        "o",
+        markers[k],
         markerfacecolor=col,
         markeredgecolor="k",
         markersize=14,
