我是 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”相关联。这可能是一个非常菜鸟的问题,如果你翻白眼的话,我很抱歉。
最佳答案
添加行
surf.set_clim([np.min(Z),np.max(Z)])
在添加颜色条之前。
3D 绘图似乎没有考虑掩蔽,因此您在数据中包含 NaN
,这会混淆自动颜色限制。
关于python - matplotlib 颜色图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/13141514/