coding180 icon
Coding180
Python Graphs: graph representations, traversal algorithms, shortest paths: examples and exercises

Python Graphs: graph representations, traversal algorithms, shortest paths: examples and exercises

Home > Tutorials > Python for beginners > Data structure


A graph is a structure that represents connections between objects. Real-world networks such as social networks, transportation systems, and computer networks can be modeled using graphs.

In Python, we can represent a graph using either an adjacency matrix or an adjacency list. An adjacency matrix is a 2D array that represents the connections between nodes. An adjacency list, on the other hand, is a dictionary where each key represents a node, and the corresponding value is a list of nodes that are adjacent to it.

Here's an example of how to create an adjacency list for a simple graph:

graph = { 
    'A': ['B', 'C'], 
    'B': ['A', 'C', 'D'], 
    'C': ['A', 'B', 'D'], 
    'D': ['B', 'C'], 
}

Graph Traversal Algorithms

Graph traversal algorithms are used to visit all the nodes in a graph. There are two main approaches: breadth-first search (BFS) and depth-first search (DFS).

In BFS, we start at the root node and explore all the neighbors before moving on to the next level of nodes. This algorithm is useful for finding the shortest path between two nodes.

In DFS, we start at the root node and explore as far as possible along each branch before backtracking. This algorithm is useful for finding cycles and paths that are not necessarily the shortest.

Here's an example of how to implement BFS in Python:

def bfs(graph, start_node):
    visited = []
    queue = [start_node]
    
    while queue:
        node = queue.pop(0)
        if node not in visited:
            visited.append(node)
            neighbors = graph[node]
            for neighbor in neighbors:
                queue.append(neighbor)
    return visited

Shortest Paths

Shortest path algorithms are used to find the shortest path between two nodes in a graph. The most commonly used algorithm for this is Dijkstra's algorithm.

Dijkstra's algorithm works by maintaining a list of tentative distances to all nodes and continually updating them as it explores the graph. Once it has visited all the nodes, it returns the shortest path.

Here's an example of how to implement Dijkstra's algorithm in Python:

import heapq

def dijkstra(graph, start_node):
    distances = {node: float('inf') for node in graph}
    distances[start_node] = 0
    pq = [(0, start_node)]
    
    while pq:
        current_distance, current_node = heapq.heappop(pq)
        
        if current_distance > distances[current_node]:
            continue
        
        for neighbor, weight in graph[current_node].items():
            distance = current_distance + weight
            if distance < distances[neighbor]:
                distances[neighbor] = distance
                heapq.heappush(pq, (distance, neighbor))
    
    return distances

Examples and Exercises

Example 1

Given the following graph, write a program to find the shortest path between A and D.

     2
A ------ B
|\       | \
| \      |  1
|  \     |   \
|   \    |    C
|    \   |   /
|     \  | 2
|      \ |/
D ------ E
     3

Example 2

Given the following graph, write a program to find the shortest path between A and F.

A ----- B
|\      | \
| \     |  4
|  2    |   \
|   \   |    C
|    \  |   /
|     \ | 3
D ----- E - F
         1

Exercise

Create a Python function that takes in a graph represented as an adjacency list and returns the minimum spanning tree using Kruskal's algorithm.

Conclusion

In this article, we introduced you to graphs and showed you how to represent them in Python. We also covered different algorithms for traversing graphs and finding the shortest paths between nodes. With the examples and exercises provided, you should now have a good understanding of graphs and how they can be used in computer science.


user

Robort Gabriel

Lagos, Nigeria

Freelance Web Developer, Native Android Developer, and Coding Tutor.

Skills
  • UI / UX
  • JavaScript
  • Python
  • PHP
  • Kotlin
  • Java
  • Bootrap
  • Android
  • Laravel