from random import shuffle

# 行數(shù)、列數(shù)、地雷數(shù)
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)]

# 打亂序列順序，達(dá)到隨機的效果
shuffle(ls)

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

# 對沒有雷的格子填充數(shù)字，該數(shù)字表示周圍地雷總數(shù)，注意邊緣格子，不要越界
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)

# 最終結(jié)果展示
[print(v) for v in matrix]

import numpy as np

# 行數(shù)、列數(shù)、地雷數(shù)
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)

# 打亂序列順序，達(dá)到隨機的效果
np.random.shuffle(ls)
print(ls)

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

# 對沒有雷的格子填充數(shù)字，該數(shù)字表示周圍地雷總數(shù)，注意邊緣格子，不要越界
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)

# 最終結(jié)果展示
print(matrix)