Matplotlib中心plot_surface在轴区域

时间:2017-06-09 19:56:58

标签: python matplotlib plot surface

我正在尝试创建一个三维曲面图但结果未与原点对齐: enter image description here

您可以在Z投影上轻松看到它在X轴和Y轴上有点偏离。我的数据是一个矩阵,其中前两列是我的X和Y,第三列是我的Z.这是我计算值的方式:

bins = np.linspace(0, 1, 1000)
indices = np.digitize(data[:, :2], bins, right=True)
X, Y = np.meshgrid(bins, bins, indexing='ij')
Q = np.zeros_like(X)
C = np.zeros_like(X)
d = data[:, 2]
for idx in range(0, len(indices)):
    (i,j) = indices[idx]
    Q[i, j] += d[idx]
    C[i, j] += 1
C[C == 0] = 1 # avoid division by zero errors
Q /= C
return (X, Y, Q, C)

绘图代码:

fig = plt.figure(figsize=(30, 15))
ax = fig.add_subplot(121, projection='3d')
ax.set_xlim3d(-0.5, 1)
ax.set_ylim3d(0, 1.5)
ax.set_zlim3d(-1.5, 1)
Q = ndi.gaussian_filter(Q, sigma=10)
ax.plot_surface(X, Y, Q, cmap=cm.coolwarm, lw=lw, rstride=rs, cstride=cs, alpha=a, antialiased=True)
ax.contourf(X, Y, Q, zdir='z', offset=-1.5, cmap=cm.coolwarm)
ax.contourf(X, Y, Q, zdir='x', offset=-0.5, cmap=cm.coolwarm)
ax.contourf(X, Y, Q, zdir='y', offset=+1.5, cmap=cm.coolwarm)

我的问题是,我是否缺少任何计算以确保曲面与原点对齐?感谢。

1 个答案:

答案 0 :(得分:2)

表面 与原点对齐。但起源是“悬在空中”#34;或者换句话说,底部窗格在contourf图下面略微(~0.05个单位),这使得显示,好像图是未对齐的。

您可能需要查看以下问题:Removing axes margins in 3D plot使用两个答案中的一个让窗格正确对齐。