1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
"""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()
|
