how to plot 3D mesh to depth map

Question

I have a 3d triangle mesh

from torch_geometric.data import Data

want to plot it as depth map from top view, so the pixel value of (x,y) equals depth value.

there is a piece of code to plot, but I don't know how to calculate z value for each pair of (x,y)

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

Answer 1

You probably want to use tripcolor .

Simply cut off the z -component from your data and use z as the value in your plot:

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