如何通过matplotlib绘制四面体网格？

Question

我想用matplotlib绘制四面体网格，以下是一个简单的四面体网格：

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)

当然，在实际应用中，网格中的四面体数量可能很大。我在谷歌找不到任何有用的帮助信息。那么用matplotlib绘制四面体网格的更好方法是什么？

此外，我可以获得网格的所有三角形面。

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)

Answer 1

matplotlib可能是该任务的错误工具。 3D绘图很难，例如，ParaView可以试用。您可以使用meshio（我的项目）将网格编写为适当的数据类型：

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

Answer 2

可以使用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()