Most examples work across multiple plotting backends, this example is also available for:
In [ ]:
import numpy as np
import networkx as nx
import holoviews as hv
from holoviews import dim, opts
from holoviews.element.graphs import layout_nodes
from bokeh.sampledata.airport_routes import routes, airports
hv.extension('matplotlib')
hv.output(fig='svg', size=300)
In [ ]:
# Create dataset indexed by AirportID and with additional value dimension
airports = hv.Dataset(airports, ['AirportID'], ['Name', 'IATA', 'City'])
source_airports = list(airports.select(City='New York').data.AirportID)
# Add connections count to routes then aggregate and select just routes connecting to NYC
routes['connections'] = 1
nyc_graph = hv.Graph((routes, airports), ['SourceID', "DestinationID"], ['connections'], label='NYC Airport Connections')\
.aggregate(function=np.count_nonzero).select(SourceID=source_airports)
# Lay out graph weighting and weight by the number of connections
np.random.seed(14)
graph = layout_nodes(nyc_graph, layout=nx.layout.fruchterman_reingold_layout, kwargs={'weight': 'connections'})
labels = hv.Labels(graph.nodes, ['x', 'y'], ['IATA', 'City'])
In [ ]:
nyc_labels = labels.select(City='New York').opts(
color='white', yoffset=0.05, size=16)
other_labels = labels[labels['City']!='New York'].opts(
color='white', size=8)
cmap = {3697: 'red', 3797: 'blue'}
(graph * nyc_labels * other_labels).opts(
opts.Graph(
bgcolor='gray', xaxis=None, yaxis=None, padding=0.1,
edge_color=dim('SourceID').categorize(cmap, 'gray'),
node_color=dim('index').categorize(cmap, 'gray'),
title='NYC Airport Connections')
)