add demo

aswarner · aswarner · commit a3d8ccd29523 · 2022-04-06T15:33:00.000-04:00
diff --git a/examples/subplots_axes_and_figures/align_titles_demo.py b/examples/subplots_axes_and_figures/align_titles_demo.py
@@ -0,0 +1,26 @@
+"""
+===============
+Aligning Labels
+===============
+
+Aligning titles using `.Figure.align_titles`.
+
+Note that the title "Title 1" would normally be much closer to the
+figure's axis.
+"""
+import matplotlib.pyplot as plt
+fig, axs = plt.subplots(2, 2,
+                        subplot_kw={"xlabel": "x", "ylabel": "", "title": "t"})
+print(axs.shape)
+axs[0][0].imshow(plt.np.zeros((3, 5)))
+axs[0][1].imshow(plt.np.zeros((5, 3)))
+axs[1][0].imshow(plt.np.zeros((1, 2)))
+axs[1][1].imshow(plt.np.zeros((2, 1)))
+axs[0][0].set_title('t2')
+rowspan1 = axs[0][0].get_subplotspec().rowspan
+print(rowspan1, rowspan1.start, rowspan1.stop)
+rowspan2 = axs[1][1].get_subplotspec().rowspan
+print(rowspan2, rowspan2.start, rowspan2.stop)
+
+fig.align_titles()
+plt.show()
