DOC: switch to OO interface in API examples

phobson · phobson · commit 3a41957964b2 · 2016-09-09T08:39:40.000-07:00
diff --git a/examples/api/bbox_intersect.py b/examples/api/bbox_intersect.py
@@ -3,9 +3,13 @@
 from matplotlib.transforms import Bbox
 from matplotlib.path import Path
 
-rect = plt.Rectangle((-1, -1), 2, 2, facecolor="#aaaaaa")
-plt.gca().add_patch(rect)
-bbox = Bbox.from_bounds(-1, -1, 2, 2)
+left, bottom, width, height = (-1, -1, 2, 2)
+rect = plt.Rectangle((left, bottom), width, height, facecolor="#aaaaaa")
+
+fig, ax = plt.subplots()
+ax.add_patch(rect)
+
+bbox = Bbox.from_bounds(left, bottom, width, height)
 
 for i in range(12):
     vertices = (np.random.random((2, 2)) - 0.5) * 6.0
@@ -14,6 +18,6 @@
         color = 'r'
     else:
         color = 'b'
-    plt.plot(vertices[:, 0], vertices[:, 1], color=color)
+    ax.plot(vertices[:, 0], vertices[:, 1], color=color)
 
 plt.show()
diff --git a/examples/api/legend_demo.py b/examples/api/legend_demo.py
@@ -7,11 +7,12 @@
 d = c[::-1]
 
 # Create plots with pre-defined labels.
-plt.plot(a, c, 'k--', label='Model length')
-plt.plot(a, d, 'k:', label='Data length')
-plt.plot(a, c + d, 'k', label='Total message length')
+fig, ax = plt.subplots()
+ax.plot(a, c, 'k--', label='Model length')
+ax.plot(a, d, 'k:', label='Data length')
+ax.plot(a, c + d, 'k', label='Total message length')
 
-legend = plt.legend(loc='upper center', shadow=True, fontsize='x-large')
+legend = ax.legend(loc='upper center', shadow=True, fontsize='x-large')
 
 # Put a nicer background color on the legend.
 legend.get_frame().set_facecolor('#00FFCC')
diff --git a/examples/api/patch_collection.py b/examples/api/patch_collection.py
@@ -42,6 +42,6 @@
 p = PatchCollection(patches, alpha=0.4)
 p.set_array(np.array(colors))
 ax.add_collection(p)
-plt.colorbar(p)
+fig.colorbar(p, ax=ax)
 
 plt.show()
diff --git a/examples/api/power_norm_demo.py b/examples/api/power_norm_demo.py
@@ -3,24 +3,23 @@
 import numpy as np
 from numpy.random import multivariate_normal
 
-data = np.vstack([multivariate_normal([10, 10], [[3, 2], [2, 3]], size=100000),
-                  multivariate_normal([30, 20], [[2, 3], [1, 3]], size=1000)
-                  ])
+data = np.vstack([
+    multivariate_normal([10, 10], [[3, 2], [2, 3]], size=100000),
+    multivariate_normal([30, 20], [[2, 3], [1, 3]], size=1000)
+])
 
 gammas = [0.8, 0.5, 0.3]
-xgrid = np.floor((len(gammas) + 1.) / 2)
-ygrid = np.ceil((len(gammas) + 1.) / 2)
 
-plt.subplot(xgrid, ygrid, 1)
-plt.title('Linear normalization')
-plt.hist2d(data[:, 0], data[:, 1], bins=100)
+fig, axes = plt.subplots(nrows=2, ncols=2)
 
-for i, gamma in enumerate(gammas):
-    plt.subplot(xgrid, ygrid, i + 2)
-    plt.title('Power law\n$(\gamma=%1.1f)$' % gamma)
-    plt.hist2d(data[:, 0], data[:, 1],
-               bins=100, norm=mcolors.PowerNorm(gamma))
+axes[0, 0].set_title('Linear normalization')
+axes[0, 0].hist2d(data[:, 0], data[:, 1], bins=100)
 
-plt.tight_layout()
+for ax, gamma in zip(axes.flat[1:], gammas):
+    ax.set_title('Power law $(\gamma=%1.1f)$' % gamma)
+    ax.hist2d(data[:, 0], data[:, 1],
+              bins=100, norm=mcolors.PowerNorm(gamma))
+
+fig.tight_layout()
 
 plt.show()
diff --git a/examples/api/radar_chart.py b/examples/api/radar_chart.py
@@ -169,14 +169,14 @@ def example_data():
     data = example_data()
     spoke_labels = data.pop(0)
 
-    fig = plt.figure(figsize=(9, 9))
+    fig, axes = plt.subplots(figsize=(9, 9), nrows=2, ncols=2,
+                             subplot_kw=dict(projection='radar'))
     fig.subplots_adjust(wspace=0.25, hspace=0.20, top=0.85, bottom=0.05)
 
     colors = ['b', 'r', 'g', 'm', 'y']
     # Plot the four cases from the example data on separate axes
-    for n, (title, case_data) in enumerate(data):
-        ax = fig.add_subplot(2, 2, n + 1, projection='radar')
-        plt.rgrids([0.2, 0.4, 0.6, 0.8])
+    for ax, (title, case_data) in zip(axes.flatten(), data):
+        ax.set_rgrids([0.2, 0.4, 0.6, 0.8])
         ax.set_title(title, weight='bold', size='medium', position=(0.5, 1.1),
                      horizontalalignment='center', verticalalignment='center')
         for d, color in zip(case_data, colors):
@@ -185,11 +185,12 @@ def example_data():
         ax.set_varlabels(spoke_labels)
 
     # add legend relative to top-left plot
-    plt.subplot(2, 2, 1)
+    ax = axes[0, 0]
     labels = ('Factor 1', 'Factor 2', 'Factor 3', 'Factor 4', 'Factor 5')
-    legend = plt.legend(labels, loc=(0.9, .95), labelspacing=0.1)
-    plt.setp(legend.get_texts(), fontsize='small')
+    legend = ax.legend(labels, loc=(0.9, .95), labelspacing=0.1, fontsize='small')
+
+    fig.text(0.5, 0.965, '5-Factor Solution Profiles Across Four Scenarios',
+             horizontalalignment='center', color='black', weight='bold',
+             size='large')
 
-    plt.figtext(0.5, 0.965, '5-Factor Solution Profiles Across Four Scenarios',
-                ha='center', color='black', weight='bold', size='large')
     plt.show()
