coolwarm is specifically designed for diverging data on shaded 3D surfaces

endolith · endolith · commit d14700892d94 · 2012-06-05T11:55:29.000-04:00
diff --git a/examples/mplot3d/contour3d_demo.py b/examples/mplot3d/contour3d_demo.py
@@ -4,7 +4,7 @@
 fig = plt.figure()
 ax = fig.add_subplot(111, projection='3d')
 X, Y, Z = axes3d.get_test_data(0.05)
-cset = ax.contour(X, Y, Z)
+cset = ax.contour(X, Y, Z, cmap = cm.coolwarm)
 ax.clabel(cset, fontsize=9, inline=1)
 
 plt.show()
diff --git a/examples/mplot3d/contour3d_demo2.py b/examples/mplot3d/contour3d_demo2.py
@@ -4,7 +4,7 @@
 fig = plt.figure()
 ax = fig.gca(projection='3d')
 X, Y, Z = axes3d.get_test_data(0.05)
-cset = ax.contour(X, Y, Z, extend3d=True)
+cset = ax.contour(X, Y, Z, extend3d=True, cmap = cm.coolwarm)
 ax.clabel(cset, fontsize=9, inline=1)
 
 plt.show()
diff --git a/examples/mplot3d/contour3d_demo3.py b/examples/mplot3d/contour3d_demo3.py
@@ -5,9 +5,9 @@
 ax = fig.gca(projection='3d')
 X, Y, Z = axes3d.get_test_data(0.05)
 ax.plot_surface(X, Y, Z, rstride=8, cstride=8, alpha=0.3)
-cset = ax.contour(X, Y, Z, zdir='z', offset=-100)
-cset = ax.contour(X, Y, Z, zdir='x', offset=-40)
-cset = ax.contour(X, Y, Z, zdir='y', offset=40)
+cset = ax.contour(X, Y, Z, zdir='z', offset=-100, cmap = cm.coolwarm)
+cset = ax.contour(X, Y, Z, zdir='x', offset=-40, cmap = cm.coolwarm)
+cset = ax.contour(X, Y, Z, zdir='y', offset=40, cmap = cm.coolwarm)
 
 ax.set_xlabel('X')
 ax.set_xlim(-40, 40)
diff --git a/examples/mplot3d/contourf3d_demo.py b/examples/mplot3d/contourf3d_demo.py
@@ -4,7 +4,7 @@
 fig = plt.figure()
 ax = fig.gca(projection='3d')
 X, Y, Z = axes3d.get_test_data(0.05)
-cset = ax.contourf(X, Y, Z)
+cset = ax.contourf(X, Y, Z, cmap = cm.coolwarm)
 ax.clabel(cset, fontsize=9, inline=1)
 
 plt.show()
diff --git a/examples/mplot3d/contourf3d_demo2.py b/examples/mplot3d/contourf3d_demo2.py
@@ -10,9 +10,9 @@
 ax = fig.gca(projection='3d')
 X, Y, Z = axes3d.get_test_data(0.05)
 ax.plot_surface(X, Y, Z, rstride=8, cstride=8, alpha=0.3)
-cset = ax.contourf(X, Y, Z, zdir='z', offset=-100)
-cset = ax.contourf(X, Y, Z, zdir='x', offset=-40)
-cset = ax.contourf(X, Y, Z, zdir='y', offset=40)
+cset = ax.contourf(X, Y, Z, zdir='z', offset=-100, cmap = cm.coolwarm)
+cset = ax.contourf(X, Y, Z, zdir='x', offset=-40, cmap = cm.coolwarm)
+cset = ax.contourf(X, Y, Z, zdir='y', offset=40, cmap = cm.coolwarm)
 
 ax.set_xlabel('X')
 ax.set_xlim(-40, 40)
diff --git a/examples/mplot3d/subplot3d_demo.py b/examples/mplot3d/subplot3d_demo.py
@@ -17,7 +17,7 @@
 X, Y = np.meshgrid(X, Y)
 R = np.sqrt(X**2 + Y**2)
 Z = np.sin(R)
-surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.jet,
+surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm,
         linewidth=0, antialiased=False)
 ax.set_zlim3d(-1.01, 1.01)
 
diff --git a/examples/mplot3d/surface3d_demo.py b/examples/mplot3d/surface3d_demo.py
@@ -11,7 +11,7 @@
 X, Y = np.meshgrid(X, Y)
 R = np.sqrt(X**2 + Y**2)
 Z = np.sin(R)
-surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.jet,
+surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm,
         linewidth=0, antialiased=False)
 ax.set_zlim(-1.01, 1.01)
 
