From 4ed9330dc0da5963124cac1b7e7caebc0eed73cd Mon Sep 17 00:00:00 2001 From: Marc Glisse Date: Sat, 16 Nov 2019 17:50:08 +0100 Subject: Draw vertex/edge/triangle with matplotlib --- src/python/example/plot_simplex_tree_dim012.py | 26 +++++++++++++++++++++++--- 1 file changed, 23 insertions(+), 3 deletions(-) (limited to 'src') diff --git a/src/python/example/plot_simplex_tree_dim012.py b/src/python/example/plot_simplex_tree_dim012.py index e6a1a4e0..f79c359b 100755 --- a/src/python/example/plot_simplex_tree_dim012.py +++ b/src/python/example/plot_simplex_tree_dim012.py @@ -9,17 +9,37 @@ cplx=gudhi.SimplexTree() cplx.insert([1,2,3,5]) cplx.insert([4,6]) cplx.insert([0]) +triangles = np.array([s[0] for s in cplx.get_skeleton(2) if len(s[0])==3]) +## With matplotlib +from mpl_toolkits.mplot3d import Axes3D +from mpl_toolkits.mplot3d.art3d import Line3DCollection +import matplotlib.pyplot as plt +fig = plt.figure() +ax = fig.gca(projection='3d') +# Plot triangles +ax.plot_trisurf(points[:,0], points[:,1], points[:,2], triangles=triangles) +# Plot points +ax.scatter3D(points[:,0], points[:,1], points[:,2]) +# Plot edges +edges=[] +for s in cplx.get_skeleton(1): + e = s[0] + if len(e) == 2: + edges.append(points[[e[0],e[1]]]) +ax.add_collection3d(Line3DCollection(segments=edges)) +plt.show() + +## With mayavi from mayavi import mlab # Plot triangles -triangles = np.array([s[0] for s in cplx.get_skeleton(2) if len(s[0])==3]) mlab.triangular_mesh(points[:,0], points[:,1], points[:,2], triangles); +# Plot points +mlab.points3d(points[:,0], points[:,1], points[:,2]) # Plot edges for s in cplx.get_skeleton(1): e = s[0] if len(e) == 2: pts = points[[e[0],e[1]]] mlab.plot3d(pts[:,0],pts[:,1],pts[:,2],tube_radius=None) -# Plot points -mlab.points3d(points[:,0], points[:,1], points[:,2]) mlab.show() -- cgit v1.2.3