diff options
author | Marc Glisse <marc.glisse@inria.fr> | 2019-11-16 17:50:08 +0100 |
---|---|---|
committer | Marc Glisse <marc.glisse@inria.fr> | 2019-11-16 17:50:08 +0100 |
commit | 4ed9330dc0da5963124cac1b7e7caebc0eed73cd (patch) | |
tree | 5809692ef3773a8f3043df5e22eaea294a5ef412 /src | |
parent | eae8673750c74d1f1023862703e350b26cf35912 (diff) |
Draw vertex/edge/triangle with matplotlib
Diffstat (limited to 'src')
-rwxr-xr-x | src/python/example/plot_simplex_tree_dim012.py | 26 |
1 files changed, 23 insertions, 3 deletions
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() |