In this lab you'll take the concepts from the previous lesson and code them from scratch yourself! You'll start by unpacking Dijkstra's algorithm and write an implementation to find the shortest path between two nodes. From there, you'll expand on the initial function in order to return the path itself and create a visualization to better understand the underlying process.
In this lab you will:
- Code Dijkstra's algorithm from scratch
- Calculate simple paths and shortest paths for undirected, directed and weighted graphs
- Visualize network graphs with networkx
import networkx as nx
import matplotlib.pyplot as plt
import numpy as np
%matplotlib inline
To get started, here's the network from the previous lesson:
G = nx.navigable_small_world_graph(3, seed=3)
G = nx.relabel_nodes(G, dict(zip(G.nodes, ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I'])))
nx.draw(G, pos=nx.random_layout(G, seed=9), with_labels=True, node_color='#1cf0c7',
node_size=500, font_weight='bold', width=2, alpha=0.8)
As a helpful reference, here's the description of Dijkstra's Algorithm from the previous lesson:
Dijkstra's algorithm is essentially a depth based search. It commences at the starting node, spanning out to neighboring nodes and in turn visiting their neighbors in search of the destination. More formally, here's a general pseudocode outline for the algorithm:
- Mark all nodes as unvisited
- Set the distance of the starting node as 0, and
$\infty$ for all other nodes - Set the starting node as the current node
- Visit each of the neighbors of the current node
- For each neighbor, calculate the distance to that node traveling through the current node
- If this distance is less then the current distance recorded for that node, update the record accordingly
- Mark the current node as "visited"
- Of the unvisited nodes, set the one with the smallest distance to the current node
- Repeat steps 4 through 6 until one of the following:
- The algorithm terminates when the destination node is the current node
- Alternatively, if the smallest distance of the unvisited nodes is
$\infty$ , then no path exists to the destination node.
Note: Dijkstra's algorithm (and NetworkX's implementations demonstrated above) returns a single path. In many cases, there may be multiple paths which are tied for the shortest distance between two nodes. In such cases, it is arbitrary which path is returned.
Implement a function to perform Dijkstra's algorithm. The function should take in three parameters: the graph G
, the starting node u
, and the destination node v
. From there, return the minimum distance between nodes u
and v
. In the case that there is no path connecting u
and v
, printout an appropriate notification and return a missing value (np.nan
).
def dijkstra(G, u, v):
"""
G is the graph in question
u is the starting node
v is the destination node
"""
pass
Test out your function on a couple of node pairs, and compare the output to that of NetworkX's built in implementation to verify your results.
# Test your function here
# Compare to NetworkX's built in method
Now, update your algorithm to not only return the minimum distance between the two points, but a list of nodes outlining the directions for the shortest path from the starting node to the destination node. The direction list of nodes should start with the starting node, and end with the destination node, with intermediate step nodes in between. For example, just like the built in method from NetworkX:
nx.dijkstra_path(G, 'I', 'A')
should return
['I', 'G', 'C', 'A']
def dijkstra(G, u, v, return_path_directions=True):
"""
G is the graph in question
u is the starting node
v is the destination node
"""
Now check your updated function against the built in methods from NetworkX again.
# Your code here
Modify your function in order to produce successive plots to visualize the process of Dijkstra's algorithm. Plot the edges connecting the starting node to "visited" nodes using one color (ideally, '#00b3e6'
). Then, plot the current edge connecting the current node to the neighbor being explored with another color (ideally, '#ffd43d'
). Create a subplot of these graphs to show the process of the algorithm.
Putting these plots together, you can also create a fun interactive visual of Dijkstra's algorithm!
Here's what the search for the shortest path between F
and G
looks like:
Or broken apart, here are the individual steps shown as subplots:
def dijkstra(G, u, v, return_path_directions=True, show_plots=True):
"""
G is the graph in question
u is the starting node
v is the destination node
Returns path, distance
"""
dijkstra(G, 'F', 'G')
Well done! In this lab, you deconstructed Dijkstra's algorithm and coded your own implementation! Finding the shortest path between nodes is a foundational concept in network theory and will help inform future concepts such as centrality and betweenness in order to further analyze the structures of graphs.