add divider and annotation.

kostyafarber · kostyafarber · commit 3f741a176070 · 2022-09-27T15:31:52.000+01:00
diff --git a/examples/style_sheets/style_sheets_reference.py b/examples/style_sheets/style_sheets_reference.py
@@ -12,6 +12,7 @@
 import numpy as np
 import matplotlib.pyplot as plt
 import matplotlib.colors as mcolors
+from matplotlib.patches import Rectangle
 
 # Fixing random state for reproducibility
 np.random.seed(19680801)
@@ -65,6 +66,14 @@ def plot_colored_circles(ax, prng, nb_samples=15):
     for sty_dict, j in zip(plt.rcParams['axes.prop_cycle'], range(nb_samples)):
         ax.add_patch(plt.Circle(prng.normal(scale=3, size=2),
                                 radius=1.0, color=sty_dict['color']))
+    
+    ax.grid(visible=True)
+    # Add annotation for enabling grid
+    ax.annotate('ax.grid(True)', xy=(1, 1),
+                xytext=(5, 8),
+                va="top", ha="right",
+                )
+    
     # Force the limits to be the same across the styles (because different
     # styles may have different numbers of available colors).
     ax.set_xlim([-4, 8])
@@ -110,7 +119,7 @@ def plot_figure(style_label=""):
     prng = np.random.RandomState(96917002)
 
     fig, axs = plt.subplots(ncols=6, nrows=1, num=style_label,
-                            figsize=(14.8, 2.7), constrained_layout=True)
+                            figsize=(14.8, 2.8), constrained_layout=True)
 
     # make a suptitle, in the same style for all subfigures,
     # except those with dark backgrounds, which get a lighter color:
@@ -126,9 +135,18 @@ def plot_figure(style_label=""):
     plot_scatter(axs[0], prng)
     plot_image_and_patch(axs[1], prng)
     plot_bar_graphs(axs[2], prng)
-    plot_colored_circles(axs[3], prng)
-    plot_colored_lines(axs[4])
-    plot_histograms(axs[5], prng)
+    plot_colored_lines(axs[3])
+    plot_histograms(axs[4], prng)
+    plot_colored_circles(axs[5], prng)
+
+    # add divider
+    rec = Rectangle(
+                    (1 + 0.025, -2), 
+                    0.05, 16, 
+                    clip_on=False, 
+                    color='gray')
+                    
+    axs[4].add_artist(rec)
 
 
 if __name__ == "__main__":
