Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
Draw Networkx Graph with Pyvis
def draw_graph3(networkx_graph,notebook=True,output_filename='graph.html',show_buttons=True,only_physics_buttons=False,
height=None,width=None,bgcolor=None,font_color=None,pyvis_options=None):
"""
This function accepts a networkx graph object,
converts it to a pyvis network object preserving its node and edge attributes,
and both returns and saves a dynamic network visualization.
Valid node attributes include:
"size", "value", "title", "x", "y", "label", "color".
(For more info: https://pyvis.readthedocs.io/en/latest/documentation.html#pyvis.network.Network.add_node)
Valid edge attributes include:
"arrowStrikethrough", "hidden", "physics", "title", "value", "width"
(For more info: https://pyvis.readthedocs.io/en/latest/documentation.html#pyvis.network.Network.add_edge)
Args:
networkx_graph: The graph to convert and display
notebook: Display in Jupyter?
output_filename: Where to save the converted network
show_buttons: Show buttons in saved version of network?
only_physics_buttons: Show only buttons controlling physics of network?
height: height in px or %, e.g, "750px" or "100%
width: width in px or %, e.g, "750px" or "100%
bgcolor: background color, e.g., "black" or "#222222"
font_color: font color, e.g., "black" or "#222222"
pyvis_options: provide pyvis-specific options (https://pyvis.readthedocs.io/en/latest/documentation.html#pyvis.options.Options.set)
"""
# import
from pyvis import network as net
# make a pyvis network
network_class_parameters = {"notebook": notebook, "height": height, "width": width, "bgcolor": bgcolor, "font_color": font_color}
pyvis_graph = net.Network(**{parameter_name: parameter_value for parameter_name, parameter_value in network_class_parameters.items() if parameter_value})
# for each node and its attributes in the networkx graph
for node,node_attrs in networkx_graph.nodes(data=True):
pyvis_graph.add_node(node,**node_attrs)
# for each edge and its attributes in the networkx graph
for source,target,edge_attrs in networkx_graph.edges(data=True):
# if value/width not specified directly, and weight is specified, set 'value' to 'weight'
if not 'value' in edge_attrs and not 'width' in edge_attrs and 'weight' in edge_attrs:
# place at key 'value' the weight of the edge
edge_attrs['value']=edge_attrs['weight']
# add the edge
pyvis_graph.add_edge(source,target,**edge_attrs)
# turn buttons on
if show_buttons:
if only_physics_buttons:
pyvis_graph.show_buttons(filter_=['physics'])
else:
pyvis_graph.show_buttons()
# pyvis-specific options
if pyvis_options:
pyvis_graph.set_options(pyvis_options)
# return and also save
return pyvis_graph.show(output_filename)
##
# For example:
##
# make a new neworkx network
import networkx as nx
G=nx.Graph()
# add nodes and edges (color can be html color name or hex code)
G.add_node('a',color='red',size=4)
G.add_node('b',color='#30a1a5',size=3)
G.add_node('c',color='green',size=1)
G.add_edge('a','b',weight=1023)
G.add_edge('a','c',weight=435)
G.add_edge('b','c',weight=100)
# draw
draw_graph3(G)
@irineos1995

This comment has been minimized.

Copy link

@irineos1995 irineos1995 commented Jan 21, 2020

hey, I might be coming late in this chat. But thanks for this code, it is really awesome.
Do you know how can I display the weights of the edges in the graph?

@ajimenezhdez

This comment has been minimized.

Copy link

@ajimenezhdez ajimenezhdez commented Apr 7, 2020

Hi!

is there any way to change the edge´s color ?

@maciejkos

This comment has been minimized.

Copy link
Owner Author

@maciejkos maciejkos commented Apr 7, 2020

@ajimenezhdez

is there any way to change the edge´s color?

If pyvis supports custom edge colors and you send me a link to documentation, I could have a look.

@irineos1995 - sorry I missed your comment!

@lucikrnic

This comment has been minimized.

Copy link

@lucikrnic lucikrnic commented Jun 14, 2021

Thank you for this code!

I was wondering what the largest number of edges in a network you ran it on was? My adaptation seems to be really struggling on 1500+ edge networks.

Do you know of an alternative to adding edge by edge to the graph as this seems to be computationally heavy for larger graphs?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment