如何绘制3D网格到深度图

问题描述

我有一个3d三角形网格

from torch_geometric.data import Data

想要从顶视图将其绘制为深度图，因此（x，y）的像素值等于深度值。

有一段代码要绘制，但是我不知道如何为每对（x，y）计算z值

plt.subplot()
x = data.x[:,1]
y = data.x[:,0]
X,Y = np.meshgrid(x,y)
plt.pcolor(data.x[:,1],data.x[:,0],2] * 255,cmap='RdBu',vmin=0,vmax=255)

解决方法

您可能想使用tripcolor。

简单地从数据中删除z分量，并使用z作为绘图中的值：

tpc = ax1.tripcolor(triang,z,shading='flat')