| 1234567891011121314151617181920212223242526272829303132333435363738394041424344 |
- """Visualize a 2D/3D network and its properties"""
- # (see also example: lineage_graph.py)
- from vedo import Points, show, sin
- from vedo.pyplot import DirectedGraph
- # Create some graph with nodes and edges
- # layouts: [2d, fast2d, clustering2d, circular, circular3d, cone, force, tree]
- g = DirectedGraph(layout='fast2d')
- ##################### Use networkx to create random nodes and edges
- # import networkx
- # G = networkx.gnm_random_graph(n=20, m=35)
- # for i, j in G.edges(): g.add_edge(j,i)
- ##################### Manually create nodes and edges
- for i in range(6): g.add_child(i) # add one child node to node i
- for i in range(3): g.add_child(i)
- for i in range(3): g.add_child(i)
- for i in range(7,9): g.add_child(i)
- for i in range(3): g.add_child(12) # add 3 children to node 12
- g.add_edge(1,16)
- ##################### build and draw
- g.build()
- graph = g[0].linewidth(4) # get the vedo 3d graph lines
- nodes = graph.vertices # get the 3d points of the nodes
- pts = Points(nodes, r=10).lighting('off')
- v1 = ['node'+str(n) for n in range(len(nodes))]
- v2 = [sin(x) for x in range(len(nodes))]
- labs1 = pts.labels(v1, scale=.02, italic=True).shift(.05,0.02,0).c('green')
- labs2 = pts.labels(v2, scale=.02, precision=3).shift(.05,-.02,0).c('red')
- # Interpolate the node value to color the edges:
- graph.cmap('viridis', v2).add_scalarbar3d(c='k')
- graph.scalarbar.shift(0.15,0,0).use_bounds(True)
- pts.cmap('viridis', v2)
- # This would colorize the edges directly with solid color based on a v3 array:
- # v3 = [sin(x) for x in range(graph.ncells)]
- # graph.cmap('jet', v3).add_scalarbar()
- show(pts, graph, labs1, labs2, __doc__, zoom='tight').close()
|
