帮我写一段在迷宫中生成怪物的代码

好的，下面是一个简单的迷宫生成算法，其中包含了生成怪物的代码：

import random

def generate_maze(rows, cols):
    maze = []
    for i in range(rows):
        row = []
        for j in range(cols):
            if i == 0 or j == 0 or i == rows-1 or j == cols-1:
                row.append(1)
            else:
                row.append(0)
        maze.append(row)

    for i in range(2, rows-2, 2):
        for j in range(2, cols-2, 2):
            maze[i][j] = 1
            directions = [(0, 2), (2, 0), (0, -2), (-2, 0)]
            random.shuffle(directions)
            for direction in directions:
                dx, dy = direction
                if maze[i+dx][j+dy] == 0:
                    maze[i+dx//2][j+dy//2] = 1
                    break

    for i in range(rows):
        for j in range(cols):
            if maze[i][j] == 0 and random.random() < 0.1:
                maze[i][j] = 2  # 2 表示怪物

    return maze

maze = generate_maze(10, 10)
for row in maze:
    print(row)

这是一个简单的深度优先搜索算法，用于生成迷宫。在生成迷宫的过程中，每当我们随机选择一个未访问过的邻居时，我们有一定的概率将其标记为怪物。在上述代码中，我们将这个概率设置为 0.1，你可以根据实际需求进行调整。