使用matplotlib进行3D密度可视化

Question

我正在尝试绘制具有相关密度的3D列。 具体来说，我有一个.txt文件，其中包含4个单独的列x, y, z, density。前3列是列的笛卡尔坐标，密度是与列的高度z处的每个横截面相关联的密度值列表。

我可以使用色彩图绘制列，如下所示

 x=np.linspace(-1, 1, 100)
 z=np.linspace(-20, 5, 50)
 Xc, Zc=np.meshgrid(x, z)
 Yc = np.sqrt(1-Xc**2)

 # Draw parameters
 rstride = 1
 cstride = 1
 surf1 = ax.plot_surface(Xc, Yc, Zc, alpha=1., rstride=rstride, cstride=cstride,antialiased=False, cmap=cm.coolwarm,linewidth=0)
 surf2 = ax.plot_surface(Xc, -Yc, Zc, alpha=1., rstride=rstride, cstride=cstride, antialiased=False, cmap=cm.coolwarm,linewidth=0)

我可以将色彩图与z

相关联

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

我想将colormap与第四列中的值相关联，同时保持柱面的绘制尺寸不变。

我很感激有关此事的任何帮助。

感谢。