マス計算の平均値算出方法について

前提・実現したいこと

N×Nに分けられたマスにおいて、それぞれランダムな数字が格納されている。仮にマスが０だった場合、周囲の値があるマスの平均値を代入することとする。

発生している問題・エラーメッセージ

現在の実装では、周囲の８つのマスに必ず値がある前提でしか対応できておらず、周囲８つのマスにも０が含まれていた場合、そのマスを除いた平均値を代入する方法ができていないのが現状です。

該当のソースコード

python
1#dot[i][j]には既に値が入っております。
2
3#セルが０だった場合の処理
4for i in range(0,N):
5    for j in range(0,N):
6        if dot[i][j] == 0:
7            if i == 0 and j == 0:#左上
8                dot[i][j] = (dot[i][j+1]+dot[i+1][j]+dot[i+1][j+1])/3
9            elif i == 0 and j == N-1:#右上
10                dot[i][j] = (dot[i][j-1]+dot[i+1][j-1]+dot[i+1][j])/3
11            elif i == N-1 and j == 0:#左下
12                dot[i][j] = (dot[i-1][j]+dot[i-1][j+1]+dot[i][j+1])/3
13            elif i == N-1 and j == N-1:#右下
14                dot[i][j] = (dot[i-1][j]+dot[i-1][j-1]+dot[i][j-1])/3
15            elif i == 0:#上辺
16                dot[i][j] = (dot[i][j-1]+dot[i+1][j-1]+dot[i+1][j]+dot[i+1][j+1]+dot[i][j+1])/5
17            elif i == N-1:#下辺
18                dot[i][j] = (dot[i][j-1]+dot[i-1][j-1]+dot[i-1][j]+dot[i-1][j+1]+dot[i][j+1])/5
19            elif j == 0:#左辺
20                dot[i][j] = (dot[i-1][j]+dot[i-1][j+1]+dot[i][j]+dot[i][j+1]+dot[i+1][j]+dot[i+1][j+1])/5
21            elif j == N-1:#右辺
22                dot[i][j] = (dot[i-1][j-1]+dot[i-1][j]+dot[i][j-1]+dot[i][j]+dot[i+1][j-1]+dot[i+1][j])/5
23            else:
24                dot[i][j] = (dot[i-1][j-1]+dot[i-1][j]+dot[i-1][j+1]+dot[i][j-1]+dot[i][j]+dot[i][j+1]+dot[i+1][j-1]+dot[i+1][j]+dot[i+1][j+1])/8
25

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

1T2R3M4

2020/12/06 05:00

https://detail.chiebukuro.yahoo.co.jp/qa/question_detail/q14235439099 以下ご対応ください。 https://teratail.com/help#posted-otherservice

toast-uz

2020/12/06 12:41

発生している問題のような状況下では、周囲の0のマスも同様な計算で0以外の数値が埋められます。その場合、計算順序によって、答えが変わってしまいますが、どのように取り決めますか？

行動規範の内容に同意します

回答1件

ベストアンサー

一旦 dot のコピーを作成してこの要素の値を参照しながら dot を更新すればやりやすいと思いますね。

Python
1import copy
2
3def get_val(dot, i, j, N):
4    if 0 <= i < N and 0 <= j < N:
5        return dot[i][j]
6    return 0
7
8tmp = copy.deepcopy(dot)
9
10for i in range(0,N):
11    for j in range(0,N):
12        if tmp[i][j] == 0:
13            vsum, cnt = 0, 0
14            for s in i-1, i, i+1:
15                for t in j-1, j, j+1:
16                    val = get_val(tmp, s, t, N)
17                    if val:
18                        cnt += 1
19                        vsum += val
20            dot[i][j] = vsum / cnt
21
22print(*dot, sep='\n')