Sync colors

benchmarks/scripts/analyze.py

@@ -13,6 +13,9 @@
 from scipy.stats import mannwhitneyu
 from scipy.stats.mstats import hdquantiles
 
+default_colors = plt.rcParams['axes.prop_cycle'].by_key()['color']
+color_cycle = itertools.cycle(default_colors)
+color_map = {}
 
 precision = 0.01
 sensitivity = 0.5
@@ -312,6 +315,9 @@ def lowland_between(mode_candidate, left_peak, right_peak):
 
 
 def hd_displot(samples, label, ax):
+    if label not in color_map:
+        color_map[label] = next(color_cycle)
+    color = color_map[label]
     widths, heights = qrde_hd(samples)
     mode_ids = extract_modes(samples)
 
@@ -335,10 +341,17 @@ def hd_displot(samples, label, ax):
     xs = xs + [max_sample]
     ys = ys + [0]
 
-    ax.plot(xs, ys, label=label)
-    ax.plot(peak_xs, peak_ys, 'o', color='red')
+    ax.fill_between(xs, ys, 0, alpha=0.4, color=color)
+
+    quartiles_of_interest = [0.25, 0.5, 0.75]
+
+    for quartile in quartiles_of_interest:
+        bin = int(quartile / precision) + 1
+        ax.plot([xs[bin], xs[bin]], [0, ys[bin]], color=color)
+
+    ax.plot(xs, ys, label=label, color=color)
+    ax.plot(peak_xs, peak_ys, 'o', color=color)
     ax.legend()
-    ax.fill_between(xs, ys, 0, alpha=0.4)
 
 
 def displot(data, ax):