Merge pull request #4639 from ericmjl/mep12-transoffset.py

jenshnielsen · jenshnielsen · commit bf3095efca08 · 2015-07-12T09:47:00.000-05:00
MEP12 on transoffset.py
diff --git a/examples/pylab_examples/transoffset.py b/examples/pylab_examples/transoffset.py
@@ -17,40 +17,40 @@
 offset.
 '''
 
-import pylab as P
+import matplotlib.pyplot as plt
+import matplotlib.transforms as mtrans
+import numpy as np
+
 from matplotlib.transforms import offset_copy
 
-X = P.arange(7)
-Y = X**2
+xs = np.arange(7)
+ys = xs**2
 
-fig = P.figure(figsize=(5, 10))
-ax = P.subplot(2, 1, 1)
+fig = plt.figure(figsize=(5, 10))
+ax = plt.subplot(2, 1, 1)
 
 # If we want the same offset for each text instance,
 # we only need to make one transform.  To get the
 # transform argument to offset_copy, we need to make the axes
 # first; the subplot command above is one way to do this.
+trans_offset = mtrans.offset_copy(ax.transData, fig=fig,
+                                  x=0.05, y=0.10, units='inches')
 
-transOffset = offset_copy(ax.transData, fig=fig,
-                          x=0.05, y=0.10, units='inches')
-
-for x, y in zip(X, Y):
-    P.plot((x,), (y,), 'ro')
-    P.text(x, y, '%d, %d' % (int(x), int(y)), transform=transOffset)
+for x, y in zip(xs, ys):
+    plt.plot((x,), (y,), 'ro')
+    plt.text(x, y, '%d, %d' % (int(x), int(y)), transform=trans_offset)
 
 
 # offset_copy works for polar plots also.
+ax = plt.subplot(2, 1, 2, polar=True)
 
-ax = P.subplot(2, 1, 2, polar=True)
-
-transOffset = offset_copy(ax.transData, fig=fig, y=6, units='dots')
-
-for x, y in zip(X, Y):
-    P.polar((x,), (y,), 'ro')
-    P.text(x, y, '%d, %d' % (int(x), int(y)),
-           transform=transOffset,
-           horizontalalignment='center',
-           verticalalignment='bottom')
+trans_offset = mtrans.offset_copy(ax.transData, fig=fig, y=6, units='dots')
 
+for x, y in zip(xs, ys):
+    plt.polar((x,), (y,), 'ro')
+    plt.text(x, y, '%d, %d' % (int(x), int(y)),
+             transform=trans_offset,
+             horizontalalignment='center',
+             verticalalignment='bottom')
 
-P.show()
+plt.show()
