本文实例为大家分享了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)
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持编程网。