This code visualizes a maze with a path marked on...

April 4, 2025 at 01:46 AM

import matplotlib.pyplot as plt
import numpy as np

def display_maze_with_path(maze):
    rows, cols = len(maze), len(maze[0])
    color_map = {
        0: "dimgray", 1: "lightgray", 3: "limegreen", 4: "tomato"
    }
    color_matrix = [[color_map[cell] for cell in row] for row in maze]
    fig, ax = plt.subplots(figsize=(8, 8))
    ax.imshow([[0] * cols for _ in range(rows)], cmap="Greys", vmin=0, vmax=4)
    for i in range(rows):
        for j in range(cols):
            ax.add_patch(plt.Rectangle((j, i), 1, 1, color=color_matrix[i][j]))
    ax.set_xticks([])
    ax.set_yticks([])
    ax.set_xlim(0, cols)
    ax.set_ylim(rows, 0)
    plt.title("Maze with Path", fontsize=18)
    plt.show()
maze = [
    [0, 1, 1, 1, 0],
    [0, 0, 0, 1, 0],
    [1, 1, 0, 1, 0],
    [1, 0, 1, 1, 1],
    [0, 0, 0, 0, 0]
]
maze[0][0] = 3
maze[-1][-1] = 4
display_maze_with_path(maze)

This code visualizes a maze with a path marked on it using matplotlib. Here's a breakdown of what it does:

Imports libraries:
- matplotlib.pyplot is used for plotting and visualization.
- numpy is imported but not utilized in this code.
Defines display_maze_with_path function:
- The function accepts a 2D list (maze) representing a grid/matrix where each value represents a type of cell. Each cell is mapped to a color based on its value using a color_map dictionary:
  - 0: "dimgray" (walls or obstacles in the maze).
  - 1: "lightgray" (paths in the maze).
  - 3: "limegreen" (starting point of the maze).
  - 4: "tomato" (endpoint of the maze).
- It constructs a corresponding color_matrix, where each cell is replaced with its assigned color.
- Sets up a plot using Matplotlib (fig, ax = plt.subplots).
- Uses ax.imshow to initialize an empty grayscale grid.
- Iterates through the matrix (maze) to draw each cell with its assigned color using plt.Rectangle.
- Removes axis ticks and sets axis limits to match the maze dimensions.
- Adds a title "Maze with Path".
Maze definition and modification:
- A 5x5 maze grid (maze) is defined, consisting of:
  - 0 (walls).
  - 1 (open paths).
- The starting point of the maze (maze[0][0]) is marked with 3 (limegreen).
- The endpoint of the maze (maze[-1][-1], i.e., the bottom-right corner) is marked with 4 (tomato).
Plot the maze:
- The display_maze_with_path function is called with the modified maze.
- The maze is displayed, where:
  - Walls are dim gray.
  - Paths are light gray.
  - The starting point is green.
  - The endpoint is red.

Visual Representation:

Example output would display a grid where:
- Top-left corner (start) is green;
- Bottom-right corner (end) is red;
- Walls and paths are styled accordingly.

Generate your own explanations

Download our vscode extension

Read other generated explanations

Built by @thebuilderjr