二次元配列の隣り合う要素の比較を効率良くしたい

Question

実現したいこと

二次元配列の隣り合う要素を比較し、隣り合う要素が存在しないものは処理をスキップしたいです。

例えば、以下のような.と#の組み合わせからなる表で#が格納された場所に対して、上下左右の要素が#だった時に、隣り合っている分だけcount ++したいとします。例えば、1行目の1列目は左と上に要素がないので、処理をスキップしたいです。

.....#.
.#.###.
.####..
###.##.
.####..

考えたコード

以下のようにすれば、実装できるのですが、もっと短く書きたいです。一度だけであれば大した長さじゃないのですが、paizaのスキルチェック問題を解いていると、このような処理をしなければいけないことが多かったので質問しました。

# metrix = "上のような表"
# 以下 metrix[i][j] に対して隣り合う要素が"#"の時だけカウント
# hは行数、wは列数

count = 0
if metrix[i][j] == "#":
    if i != 0:
        if metrix[i-1][j] == "#":
            count += 1
    if j != w - 1:
        if metrix[i][j + 1] == "#":
            count += 1
    if i != h - 1:
        if metrix[i + 1][j] == "#":
            count += 1
    if j != 0:
        if metrix[i][j-1] == "#":
            count += 1

print(count)

試したこと

try except IndexErrorを使って、indexが存在しないときはカウントせずに処理を実行。
pythonではindex = -1が配列の最後の要素を示すので、期待した結果が得られませんでした。

Comments

[ppaul]

実行すると以下のエラーになります。
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
NameError: name 'i' is not defined
動くコードに修正してください。

Answer 1

ループをそれぞれいっこ手前で止めておいて
チェックは右方向と下方向だけやればいいんじゃね?
いまのコードだと二重にカウントしますよね。
まあ、有り無しを判断するんだったら関係ないかもですけど。

Comments

[pompon]

そのやり方が使える問題も多そうです！今度それでやってみます、ありがとうございます！

Answer 2

よくやるのは「番兵」を入れる考えかたです。
以下のように1回り大きい配列にして、一番外側を無効(見やすいように「_」にしています)な値を入れます。

_________
_.....#._
_.#.###._
_.####.._
_###.##._
_.####.._
_________

こうすることで、「 if i != 0:」のような、境界であるかどうかの判定が不要になります。

Comments

[pompon]

そのやり方だと、とても楽ですね！これからそれでやっていきます！ありがとうございました！