MEP12 on tricontour_vs_griddata.py

ericmjl · ericmjl · commit e0df1ddec9ff · 2015-07-11T12:17:16.000-05:00
In particular:

1. numpy random functions are imported under `rnd`.
diff --git a/examples/pylab_examples/tricontour_vs_griddata.py b/examples/pylab_examples/tricontour_vs_griddata.py
@@ -5,24 +5,24 @@
 import matplotlib.pyplot as plt
 import matplotlib.tri as tri
 import numpy as np
-from numpy.random import uniform, seed
-from matplotlib.mlab import griddata
+import numpy.random as rnd
+import matplotlib.mlab as mlab
 import time
 
-seed(0)
+rnd.seed(0)
 npts = 200
 ngridx = 100
 ngridy = 200
-x = uniform(-2, 2, npts)
-y = uniform(-2, 2, npts)
+x = rnd.uniform(-2, 2, npts)
+y = rnd.uniform(-2, 2, npts)
 z = x*np.exp(-x**2 - y**2)
 
 # griddata and contour.
 start = time.clock()
 plt.subplot(211)
 xi = np.linspace(-2.1, 2.1, ngridx)
 yi = np.linspace(-2.1, 2.1, ngridy)
-zi = griddata(x, y, z, xi, yi, interp='linear')
+zi = mlab.griddata(x, y, z, xi, yi, interp='linear')
 plt.contour(xi, yi, zi, 15, linewidths=0.5, colors='k')
 plt.contourf(xi, yi, zi, 15, cmap=plt.cm.rainbow,
              norm=plt.Normalize(vmax=abs(zi).max(), vmin=-abs(zi).max()))
@@ -48,4 +48,4 @@
 plt.title('tricontour (%d points)' % npts)
 print('tricontour seconds: %f' % (time.clock() - start))
 
-plt.show()
+plt.show()
