Merge pull request #4948 from ericmjl/mep12-layer_images.py

WeatherGod · WeatherGod · commit 677646e4b7ab · 2015-08-18T09:30:18.000-04:00
Mep12 layer images.py
diff --git a/examples/pylab_examples/layer_images.py b/examples/pylab_examples/layer_images.py
@@ -1,20 +1,20 @@
-#!/usr/bin/env python
 """
 Layer images above one another using alpha blending
 """
 from __future__ import division
-from pylab import *
+import matplotlib.pyplot as plt
+import numpy as np
 
 
 def func3(x, y):
-    return (1 - x/2 + x**5 + y**3)*exp(-(x**2 + y**2))
+    return (1 - x/2 + x**5 + y**3)*np.exp(-(x**2 + y**2))
 
 # make these smaller to increase the resolution
 dx, dy = 0.05, 0.05
 
-x = arange(-3.0, 3.0, dx)
-y = arange(-3.0, 3.0, dy)
-X, Y = meshgrid(x, y)
+x = np.arange(-3.0, 3.0, dx)
+y = np.arange(-3.0, 3.0, dy)
+X, Y = np.meshgrid(x, y)
 
 # when layering multiple images, the images need to have the same
 # extent.  This does not mean they need to have the same shape, but
@@ -24,20 +24,19 @@ def func3(x, y):
 # interpolation edge effects
 
 
-xmin, xmax, ymin, ymax = amin(x), amax(x), amin(y), amax(y)
+xmin, xmax, ymin, ymax = np.amin(x), np.amax(x), np.amin(y), np.amax(y)
 extent = xmin, xmax, ymin, ymax
 fig = plt.figure(frameon=False)
 
-Z1 = array(([0, 1]*4 + [1, 0]*4)*4)
+Z1 = np.array(([0, 1]*4 + [1, 0]*4)*4)
 Z1.shape = (8, 8)  # chessboard
-im1 = imshow(Z1, cmap=cm.gray, interpolation='nearest',
-             extent=extent)
-hold(True)
+im1 = plt.imshow(Z1, cmap=plt.cm.gray, interpolation='nearest',
+                 extent=extent)
+plt.hold(True)
 
 Z2 = func3(X, Y)
 
-im2 = imshow(Z2, cmap=cm.jet, alpha=.9, interpolation='bilinear',
-             extent=extent)
-#axis([xmin, xmax, ymin, ymax])
+im2 = plt.imshow(Z2, cmap=plt.cm.jet, alpha=.9, interpolation='bilinear',
+                 extent=extent)
 
-show()
+plt.show()
