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

需求说明输入：行数m、列数n、地雷数k
输出：随机扫雷地图
第一步，生成地图和随机地雷
第二步，没有地雷的格子填充数字，该数字表示周边地雷总数
思路 【玩Python|Python生成扫雷地图(并非游戏实现)】
文章图片

代码实现下面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)