Question

我是python的新手，在安装之后我已经完成了使用matplotlib绘制我的3D数据。可悲的是，我唯一不知道如何完成的是颜色部分。我的图像只显示表面，但根本不使用颜色条。这是我的代码。

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
from matplotlib.mlab import griddata
import numpy as np

fig = plt.figure()
ax = fig.gca(projection='3d')

data = np.genfromtxt('Uizq.txt')
x = data[:,0]
y = data[:,1]
z = data[:,2]

xi = np.linspace(min(x), max(x))
yi = np.linspace(min(y), max(y))

ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('U')

X, Y = np.meshgrid(xi, yi)
Z = griddata(x, y, z, xi, yi)
ax.set_zlim3d(np.min(Z), np.max(Z))


surf = ax.plot_surface(X, Y, Z, rstride=2, cstride=2, cmap=cm.jet,
                   linewidth=0.5, antialiased=False)

fig.colorbar(surf, shrink=0.5, aspect=5)

plt.show()

plot http://i49.tinypic.com/263kcbq.png

你可以清楚地看到它全是蓝色的，我想使用完整的cm.jet光谱将颜色与“U”联系起来。这可能是一个非常的菜鸟问题，如果你翻白眼就很抱歉。

Answer 1

添加行

surf.set_clim([np.min(Z),np.max(Z)])

添加颜色条之前。

3D绘图似乎没有考虑到屏蔽，因此您在数据中包含NaN，这会混淆自动颜色限制。

matplotlib的色彩映射

1 个答案: