diff options
author | jarrodmillman <jarrod.millman@gmail.com> | 2022-12-14 17:21:13 +0000 |
---|---|---|
committer | jarrodmillman <jarrod.millman@gmail.com> | 2022-12-14 17:21:13 +0000 |
commit | 832c558e3507e5cb667a622b5372f91384ab026f (patch) | |
tree | 74481f14ab9cfb5c6984ab59b378cdc857a8180d /_downloads/a3194c86dd5e7ffeab14caf2b61a88b9 | |
parent | 71985f91c82bf85657dce6a74669e93ec8d29e11 (diff) | |
download | networkx-832c558e3507e5cb667a622b5372f91384ab026f.tar.gz |
Deploying to gh-pages from @ networkx/networkx@6be702047b1bb596a8010cf80911bb6ea939b1d1 🚀
Diffstat (limited to '_downloads/a3194c86dd5e7ffeab14caf2b61a88b9')
-rw-r--r-- | _downloads/a3194c86dd5e7ffeab14caf2b61a88b9/mayavi2_spring.py | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/_downloads/a3194c86dd5e7ffeab14caf2b61a88b9/mayavi2_spring.py b/_downloads/a3194c86dd5e7ffeab14caf2b61a88b9/mayavi2_spring.py new file mode 100644 index 00000000..bec09694 --- /dev/null +++ b/_downloads/a3194c86dd5e7ffeab14caf2b61a88b9/mayavi2_spring.py @@ -0,0 +1,43 @@ +""" +======= +Mayavi2 +======= + +""" + +import networkx as nx +import numpy as np +from mayavi import mlab + +# some graphs to try +# H=nx.krackhardt_kite_graph() +# H=nx.Graph();H.add_edge('a','b');H.add_edge('a','c');H.add_edge('a','d') +# H=nx.grid_2d_graph(4,5) +H = nx.cycle_graph(20) + +# reorder nodes from 0,len(G)-1 +G = nx.convert_node_labels_to_integers(H) +# 3d spring layout +pos = nx.spring_layout(G, dim=3, seed=1001) +# numpy array of x,y,z positions in sorted node order +xyz = np.array([pos[v] for v in sorted(G)]) +# scalar colors +scalars = np.array(list(G.nodes())) + 5 + +mlab.figure() + +pts = mlab.points3d( + xyz[:, 0], + xyz[:, 1], + xyz[:, 2], + scalars, + scale_factor=0.1, + scale_mode="none", + colormap="Blues", + resolution=20, +) + +pts.mlab_source.dataset.lines = np.array(list(G.edges())) +tube = mlab.pipeline.tube(pts, tube_radius=0.01) +mlab.pipeline.surface(tube, color=(0.8, 0.8, 0.8)) +mlab.orientation_axes() |