How to draw a Tetrahedron mesh by matplotlib?

5.4k views Asked by At

I want to plot a tetrahedron mesh by matplotlib, and the following are a simple tetrahedron mesh:

xyz = np.array([
    [-1,-1,-1],
    [ 1,-1,-1], 
    [ 1, 1,-1],
    [-1, 1,-1],
    [-1,-1, 1],
    [ 1,-1, 1], 
    [ 1, 1, 1],
    [-1, 1, 1]], dtype=np.float) 

tets = np.array([
    [0,1,2,6],
    [0,5,1,6],
    [0,4,5,6],
    [0,7,4,6],
    [0,3,7,6],
    [0,2,3,6]], dtype=np.int)

Of course, in practical applications, the number of tetrahedrons in a mesh can be large. I can't find any useful help information in google. So what is the better way to plot a tetrahedron mesh by matplotlib?

Furthermore, I can get all the triangle faces of the mesh.

tri = np.array([
    [0, 2, 1],
    [0, 1, 5],
    [0, 6, 1],
    [0, 3, 2],
    [0, 2, 6],
    [0, 6, 3],
    [0, 7, 3],
    [0, 5, 4],
    [0, 6, 4],
    [0, 4, 7],
    [0, 6, 5],
    [0, 6, 7],
    [1, 2, 6],
    [5, 1, 6],
    [2, 3, 6],
    [3, 7, 6],
    [4, 5, 6],
    [7, 4, 6]], dtype=np.int)
2

There are 2 answers

2
Nico Schlömer On BEST ANSWER

matplotlib is perhaps the wrong tool for the task. 3D plotting is hard, and there is, e.g., ParaView to try out. You can use meshio (a project of mine) to write your mesh to an appropriate data type:

import meshio
meshio.write('out.vtu', xyz, {'tetra': tets})

enter image description here

0
Huayi Wei On

One can use mpl_toolkits.mplot3d.art3d.Poly3DCollection:

import mpl_toolkits.mplot3d as a3
axes = a3.Axes3D(pl.figure())
vts = xyz[tri, :]
tri = a3.art3d.Poly3DCollection(vts)
tri.set_alpha(0.2)
tri.set_color('grey')
axes.add_collection3d(tri)
axes.plot(point[:,0], point[:,1], point[:,2], 'ko')
axes.set_axis_off()
axes.set_aspect('equal')
pl.show()