Python生成掃雷地圖的方法是怎樣的

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)

看完上述內容是否對您有幫助呢？如果還想對相關知識有進一步的了解或閱讀更多相關文章，請關注億速云行業資訊頻道，感謝您對億速云的支持。