Update interpolation_methods.py to use viridis

mdboom · mdboom · commit c9cae50246ac · 2016-02-17T09:13:31.000-05:00
diff --git a/examples/images_contours_and_fields/interpolation_methods.py b/examples/images_contours_and_fields/interpolation_methods.py
@@ -18,6 +18,7 @@
            'spline36', 'hanning', 'hamming', 'hermite', 'kaiser', 'quadric',
            'catrom', 'gaussian', 'bessel', 'mitchell', 'sinc', 'lanczos']
 
+np.random.seed(0)
 grid = np.random.rand(4, 4)
 
 fig, axes = plt.subplots(3, 6, figsize=(12, 6),
@@ -26,7 +27,7 @@
 fig.subplots_adjust(hspace=0.3, wspace=0.05)
 
 for ax, interp_method in zip(axes.flat, methods):
-    ax.imshow(grid, interpolation=interp_method)
+    ax.imshow(grid, interpolation=interp_method, cmap='viridis')
     ax.set_title(interp_method)
 
-plt.show()
+plt.savefig("test.png")
