diff --git a/examples/graph_visual.py b/examples/graph_visual.py index 72c72bad..b455e6de 100644 --- a/examples/graph_visual.py +++ b/examples/graph_visual.py @@ -1,5 +1,6 @@ import networkx as nx from pyvis.network import Network +import random # Load the GraphML file G = nx.read_graphml('./dickens/graph_chunk_entity_relation.graphml') @@ -10,5 +11,9 @@ net = Network(notebook=True) # Convert NetworkX graph to Pyvis network net.from_nx(G) +# Add colors to nodes +for node in net.nodes: + node['color'] = "#{:06x}".format(random.randint(0, 0xFFFFFF)) + # Save and display the network net.show('knowledge_graph.html') \ No newline at end of file