パイソン、森林火災シミュレーションの延焼ステップについて

Question

前提・実現したいこと

二次元配列を用いた森林火災のシミュレーションをプログラミングで作成しています。使用している言語はpythonです。visual studio を利用しています。
＜概要＞
N×Nマス上の地面に、木を配置していきます。確率ｐで木が生えている（状態：１）。確率１－ｐで木が生えていない（状態：０）　例：map1[x][y]=1→木が生えている　といった感じ。
燃えている木の状態を２、鎮火した状態の木を３とする。
map1[x][y]の初期値は０か１とし、最初のステップでどこかの木を２にし、完全に鎮火するまでステップを更新していく。（ｘ、ｙ）が１の時、隣接するマスのうち、どこか一つでも２の状態のマスがあれば（ｘ、ｙ）は２となり、次のステップで３となる。２が広がるのは１ステップで1マスのみとなっている。

＜問題＞
1ステップの間に2マス以上の延焼が起こってしまいます。私の方針として4隅、4辺、中といった場合分けでコードを作成しました。延焼のステップを何度か書き換えてみましたが、やはりうまくいかなかったです。

どこら辺がおかしいか教えていただけますでしょうか。
よろしくお願いします。

該当のソースコード

import random
import itertools

random.seed(50)

p = 0.7 #木の生えている確率
N = 5 #森の広さ
haji = [0,-1]
tyousei = [1,-1]

map1 = []
for x in range(N):
map1.append([0]*N)
for y in range(N):
    if random.random() <= p:
        map1[x][y] = 1

map2 = []
for _ in range (N):
map2.append([0]*N)

while True:
a = random.randint(0,N-1)
b = random.randint(0,N-1)
if map1[a][b] == 1:
    map1[a][b] = 2
    break

print(*map1, sep="\n")

all_list = itertools.chain(*map1)
all_list = list(all_list)

while 2 in all_list:
for x in haji:  #隅っこ
    if x == 0: #左端の隅っこ
        for y,z in zip(haji,tyousei):
            if map1[y][x] == 1:
                right = map1[y][x+1]
                updown = map1[y+z][x]
                if right == 2 or updown == 2:
                    map2[y][x] = 2
                else:
                    map2[y][x] = 1
            elif map1[y][x] == 2:
                map2[y][x] = 3
            else:
                map2[y][x] = map1[y][x]
    else:   #右端の隅っこ
        for y,z in zip(haji,tyousei):
            if map1[y][x] == 1:
                left = map1[y][x-1]
                updown = map1[y+z][x]
                if left == 2 or updown == 2:
                    map2[y][x] = 2
                else:
                    map2[y][x] = 1
            elif map1[y][x] == 2:
                map2[y][x] = 3
            else:
                map2[y][x] = map1[y][x]

for x,z in zip(haji,tyousei):
    for y in range(1,N-1): #上下の縁
        if map1[x][y] == 1:
            left = map1[x][y-1]
            right = map1[x][y+1]
            updown = map1[x+z][y]
            if left == 2 or right == 2 or updown == 2:
                map2[x][y] = 2
            else:
                map2[x][y] = 1
        elif map1[x][y] == 2:
            map2[x][y]=3
        else:
            map2[x][y] = map1[x][y]

for y,z in zip(haji,tyousei):
    for x in range(1,N-1): #左右の縁
        if map1[x][y] == 1:
            up = map1[x-1][y]
            down = map1[x+1][y]
            side = map1[x][y+z]
            if up == 2 or down == 2 or side == 2:
                map2[x][y] = 2  
            else:
                map2[x][y] = 1
        elif map1[x][y] == 2:
            map2[x][y] = 3
        else:
            map2[x][y] = map1[x][y]

for x in range(1, N-1):  #縁以外
    for y in range(1, N-1):
        if map1[x][y] == 1:
            up = map1[x-1][y]
            down = map1[x+1][y]
            left = map1[x][y-1]
            right = map1[x][y+1]
            if up == 2 or down == 2 or left == 2 or right == 2:
                map2[x][y] = 2    #延焼
            else:
                map2[x][y] = 1 #変化なし
        elif map1[x][y] == 2:
            map2[x][y] = 3    #鎮火
        else:
            map2[x][y] = map1[x][y]    #変化なし

map1 = map2
all_list = itertools.chain(*map1)
all_list = list(all_list)
print("============")
print(*map1, sep="\n")

試したこと

乱数の値は固定しています。

Answer 1

import copy

を追加して

- map1 = map2
+ map1 = copy.deepcopy(map2)

Comments

[RYO_A128]

解決しました！！
ありがとうございます！！