首先，我们需要导入所需的库和模块：
导入pygame库
import pygame
定义一个函数，使用Prim算法生成迷宫
def prim_maze_generator(width, height):

初始化二维数组表示迷宫

maze = [[0 for in range(height)] for in range(width)]

随机选择一个起始点

start = (randint(0, width-1), randint(0, height-1))
maze[start[0]][start[1]] = 1

定义已访问的点集合

visited = set([start])

定义迷宫的边

edges = [(0, 1), (0, -1), (1, 0), (-1, 0)]

定义方向

directions = [(0, 1), (1, 0), (0, -1), (-1, 0)]

生成迷宫

while len(visited) < width*height-1:
x, y = visited[-1]
for dx, dy in edges:
nx, ny = x+dx, y+dy
if 0 <= nx < width and 0 <= ny < height and maze[nx][ny] == 0 and (nx, ny) not in visited:
maze[x][y], maze[nx][ny] = maze[nx][ny], maze[x][y]
visited.add((nx, ny))
nx2, ny2 = x+dx2, y+dy2
if abs(dx2) > abs(dy2):
maze[x+dx2//2][y] = -1
maze[x+dx2//2+1][y] = -1
elif abs(dy2) > abs(dx2):
maze[x][y+dy2//2] = -1
maze[x][y+dy2//2+1] = -1
else:
maze[x+dx2//2][y+dy2//2] = -1
maze[x+dx2//2+1][y+dy2//2] = -1
maze[x+dx2//2][y+dy2//2+1] = -1
maze[x+dx2//2+1][y+dy2//2+1] = -1
return maze
接下来，我们使用Pygame库绘制迷宫：
初始化Pygame窗口和画布
pygame.init()
window_width = 800
window_height = 600
window = pygame.display.set_mode((window_width, window_height))
pygame.display.set_caption(‘Prim算法生成迷宫’)
定义绘制迷宫的函数
def draw_maze(maze):
for row in maze:
for cell in row:
if cell == 0:
pygame.draw.rect(window, (0, 255, 0), (randint(0, window_width-30), randint(0, window_height-30), 30, 30))
elif cell == -1:
pygame.draw.rect(window, (255, 0, 0), (randint(0, window_width-30), randint(0, window_height-30), 30, 30))
else:
pygame.draw.rect(window, (0, 0, 255), (randint(0, window_width-30), randint(0, window_height-30), 30, 30))
生成迷宫并绘制可视化结果
m = prim_maze_generator(window_width//