Python生成扫雷地图(并非游戏实现)

需求说明

输入：行数m、列数n、地雷数k
输出：随机扫雷地图

第一步，生成地图和随机地雷
第二步，没有地雷的格子填充数字，该数字表示周边地雷总数

思路

代码实现

下面2种实现，分别是不用numpy和用numpy，思路是一样的

from random import shuffle

# 行数、列数、地雷数
m, n, k = 5, 10, 15

# 地雷符号，此处用9表示
landmine = 9

# 初始化序列，总长为m*n，k前是雷，k后是0
ls = [landmine for _ in range(k)] + [0 for _ in range(m * n - k)]

# 打乱序列顺序，达到随机的效果
shuffle(ls)

# m*n的序列=>m行n列的矩阵
matrix = [[ls[i + j * n] for i in range(n)] for j in range(m)]

# 对没有雷的格子填充数字，该数字表示周围地雷总数，注意边缘格子，不要越界
for i in range(m):
    for j in range(n):
        if matrix[i][j] == 0:
            matrix[i][j] = sum(
                1 for a in range(max(i - 1, 0), min(i + 2, m))
                for b in range(max(j - 1, 0), min(j + 2, n))
                if matrix[a][b] == landmine)

# 最终结果展示
[print(v) for v in matrix]

import numpy as np

# 行数、列数、地雷数
m, n, k = 5, 10, 15

# 地雷符号，此处用9表示
landmine = 9

# 初始化总长为m*n的序列，TopK是雷，k后是0
ls = [landmine for _ in range(k)] + [0 for _ in range(m * n - k)]
print(ls)

# 打乱序列顺序，达到随机的效果
np.random.shuffle(ls)
print(ls)

# m*n的序列=>m行n列的矩阵
matrix = np.reshape(ls, (m, n))
print(matrix)

# 对没有雷的格子填充数字，该数字表示周围地雷总数，注意边缘格子，不要越界
for i in range(m):
    for j in range(n):
        if matrix[i][j] == 0:
            matrix[i][j] = sum(
                1 for a in range(max(i - 1, 0), min(i + 2, m))
                for b in range(max(j - 1, 0), min(j + 2, n))
                if matrix[a][b] == landmine)

# 最终结果展示
print(matrix)