tricontour example rainbow instead of jet, center norm

endolith · endolith · commit 04409c83ea39 · 2012-06-05T13:26:55.000-04:00
diff --git a/examples/pylab_examples/tricontour_vs_griddata.py b/examples/pylab_examples/tricontour_vs_griddata.py
@@ -24,7 +24,7 @@
 yi = np.linspace(-2.1,2.1,ngridy)
 zi = 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.jet)
+plt.contourf(xi,yi,zi,15,cmap=plt.cm.rainbow, norm = Normalize(vmax=abs(zi).max(), vmin=-abs(zi).max()))
 plt.colorbar() # draw colorbar
 plt.plot(x, y, 'ko', ms=3)
 plt.xlim(-2,2)
@@ -37,7 +37,7 @@
 plt.subplot(212)
 triang = tri.Triangulation(x, y)
 plt.tricontour(x, y, z, 15, linewidths=0.5, colors='k')
-plt.tricontourf(x, y, z, 15, cmap=plt.cm.jet)
+plt.tricontourf(x, y, z, 15, cmap=plt.cm.rainbow, norm = Normalize(vmax=abs(zi).max(), vmin=-abs(zi).max()))
 plt.colorbar()
 plt.plot(x, y, 'ko', ms=3)
 plt.xlim(-2,2)
