10. Graph Plots

[1]:

%matplotlib inline
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import seaborn as sns
import networkx as nx
import warnings
from _runtime import graph_layout

plt.style.use('ggplot')
np.random.seed(37)
warnings.filterwarnings('ignore')

10.1. Basic undirected graph

Look at the API documentation for more options.

[2]:

g = nx.Graph()

g.add_node('a')
g.add_node('b')
g.add_node('c')

g.add_edge('a', 'b')
g.add_edge('b', 'c')

fig, ax = plt.subplots(figsize=(5, 5))

nx.draw(g, with_labels=True, font_weight='bold', node_color='r', ax=ax)

10.2. Basic directed graph

[3]:

g = nx.DiGraph()

g.add_node('a')
g.add_node('b')
g.add_node('c')

g.add_edge('a', 'b')
g.add_edge('b', 'c')

fig, ax = plt.subplots(figsize=(5, 5))

nx.draw(g, with_labels=True, font_weight='bold', font_family='monospace', node_color='r', ax=ax)

10.3. Styling nodes and edges

Look at the API references below for more options.

[4]:

g = nx.DiGraph()

g.add_node('a')
g.add_node('b')
g.add_node('c')
g.add_node('d')

g.add_edge('a', 'c', weight=1)
g.add_edge('b', 'c', weight=2)
g.add_edge('c', 'd', weight=3)

fig, ax = plt.subplots(figsize=(5, 5))

pos = graph_layout(g, layout='dot')

params = {
    'node_color': 'r',
    'node_size': 350,
    'node_shape': 's',
    'alpha': 0.5,
    'pos': pos,
    'ax': ax
}
_ = nx.drawing.nx_pylab.draw_networkx_nodes(g, **params)

params = {
    'font_size': 15,
    'font_color': 'k',
    'font_family': 'monospace',
    'pos': pos,
    'ax': ax
}
_ = nx.drawing.nx_pylab.draw_networkx_labels(g, **params)

params = {
    'width': 1.5,
    'alpha': 0.5,
    'edge_color': 'b',
    'arrowsize': 20,
    'pos': pos,
    'ax': ax
}
_ = nx.drawing.nx_pylab.draw_networkx_edges(g, **params)

params = {
    'edge_labels': {e: g.edges[e]['weight'] for e in g.edges()},
    'font_family': 'monospace',
    'pos': pos,
    'ax': ax
}
_ = nx.drawing.nx_pylab.draw_networkx_edge_labels(g, **params)

10.4. Layouts with Matplotlib

[5]:

def add_edges(g, node_id, n):
    for i in range(n):
        g.add_edge(node_id, f'{node_id}{i}')

g = nx.DiGraph()

g.add_edge('a', 'c')
g.add_edge('b', 'c')
g.add_edge('c', 'd')
g.add_edge('d', 'e')

add_edges(g, 'a', 5)
add_edges(g, 'b', 5)
add_edges(g, 'c', 5)
add_edges(g, 'd', 10)

[6]:

import copy

fig, axes = plt.subplots(6, 1, figsize=(20, 30))
axes = np.ravel(axes)

params = {
    'with_labels': True,
    'font_color': 'w',
    'font_family': 'monospace',
    'font_weight': 'bold',
    'font_size': 15,
    'node_color': 'r',
    'node_shape': 's',
    'node_size': 400,
    'arrowsize': 20,
    'arrowstyle': '-|>',
    'alpha': 0.9
}

drawing_names = [
    'Circular',
    'Kamada_Kawai',
    'Planar',
    'Spectral',
    'Spring',
    'Shell'
]
drawing_functions = [
    nx.drawing.nx_pylab.draw_circular,
    nx.drawing.nx_pylab.draw_kamada_kawai,
    nx.drawing.nx_pylab.draw_planar,
    nx.drawing.nx_pylab.draw_spectral,
    nx.drawing.nx_pylab.draw_spring,
    nx.drawing.nx_pylab.draw_shell
]

for name, f, ax in zip(drawing_names, drawing_functions, axes):
    p = copy.deepcopy(params)
    p['ax'] = ax

    f(g, **p)
    _ = ax.set_title(name)

10.5. Layouts with dot

dot layouts: circo, dot, fdp, neato, nop, nop1, nop2, osage, patchwork, sfdp, twopi
colors

[7]:

from itertools import cycle

g = nx.DiGraph()

g.add_edge('a', 'c')
g.add_edge('b', 'c')
g.add_edge('c', 'd')
g.add_edge('d', 'e')

add_edges(g, 'a', 5)
add_edges(g, 'b', 5)
add_edges(g, 'c', 5)
add_edges(g, 'd', 10)

layouts = ['circo', 'dot', 'fdp', 'neato', 'osage', 'patchwork', 'sfdp', 'twopi']
positions = [graph_layout(g, layout=layout) for layout in layouts]
node_shapes = cycle('so^v><dp')
node_colors = cycle('bgrcmy')
edge_colors = cycle('ymcrgb')
font_colors = 'wkkkkkkk'

fig, axes = plt.subplots(len(layouts), 1, figsize=(15, 50))
axes = np.ravel(axes)

for layout, pos, node_shape, node_color, edge_color, font_color, ax in zip(layouts, positions, node_shapes, node_colors, edge_colors, font_colors, axes):
    params = {
        'node_color': node_color,
        'node_size': 450,
        'node_shape': node_shape,
        'alpha': 0.5,
        'pos': pos,
        'ax': ax
    }
    _ = nx.drawing.nx_pylab.draw_networkx_nodes(g, **params)

    params = {
        'font_size': 15,
        'font_color': font_color,
        'font_family': 'monospace',
        'pos': pos,
        'ax': ax
    }
    _ = nx.drawing.nx_pylab.draw_networkx_labels(g, **params)

    params = {
        'width': 1.5,
        'alpha': 0.5,
        'edge_color': edge_color,
        'arrowsize': 20,
        'pos': pos,
        'ax': ax
    }
    _ = nx.drawing.nx_pylab.draw_networkx_edges(g, **params)

    _ = ax.set_title(f'layout={layout}, node_shape={node_shape}')
    _ = ax.axes.get_xaxis().set_visible(False)
    _ = ax.axes.get_yaxis().set_visible(False)