Matplotlib surface plot showing same color for different values

Question

Why do all points get the same value? I would like color to vary with energy.

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
from matplotlib import cm
from numpy import *

x = linspace(0.2, 2, 11)
y = linspace(0.1, 1, 11)
alpha, beta = meshgrid(x,y)
energy = matrix(loadtxt('energyPlotfileN6.txt'))

fig = plt.figure()
ax = fig.gca(projection='3d')
ax.plot_surface(alpha, beta, energy, cmap='summer', vmin=energy.min(), vmax=energy.max())
plt.xlabel("alpha")
plt.ylabel("beta")
ax.set_zlabel("energy")
plt.show()

The result is shown below

Surface plot without desired colour mapping

Do you get the same output if you replace cmap='summer' with cmap=cm.summer? — jonchar
– jonchar, Commented May 23, 2016 at 17:44

blaylockbk · Accepted Answer · 2017-07-13 16:57:39Z

2

Used the answer from above (add the cstride and rstride arguments), but wanted to add a visualization of the difference...

In my case I'm plotting terrain...

No stride:

surf = ax.plot_surface(topo['lon'], topo['lat'], topo['value'],
                       cmap='terrain', vmax=2800, vmin=1300,
                       linewidth=.1, antialiased=False)

With stride:

surf = ax.plot_surface(topo['lon'], topo['lat'], topo['value'],
                           cmap='terrain', vmax=2800, vmin=1300,
                           linewidth=.1, antialiased=False,
                           rstride=1, cstride=1)

answered Jul 13, 2017 at 16:57

blaylockbk

3,5212 gold badges34 silver badges56 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Han-Kwang Nienhuys · Accepted Answer · 2016-05-23 17:47:27Z

1

ax.plot_surface(alpha, beta, energy, cstride=1, rstride=1, cmap='summer', vmin=energy.min(), vmax=energy.max())

Note the cstride and rstride parameters.

Axes3D.plot_surface documentation.

answered May 23, 2016 at 17:47

Han-Kwang Nienhuys

3,3242 gold badges15 silver badges33 bronze badges

1 Comment

blaylockbk Over a year ago

Amen to using the cstride and rstride parameters!

Collectives™ on Stack Overflow

Matplotlib surface plot showing same color for different values

2 Answers 2

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related