matplotlib 颜色图

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()

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

Answer 1

添加一行

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

在添加彩条之前。

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