python3 2次元リスト インデックス 検索

Question

下記の2次元リストで、並びが一致しているところを探してインデックスを出力したいのですが、良い方法はありますでしょうか？

『対象リスト』
[[0, 0, 0,],
[1, 0, 1,],
[0, 1, 0,],
[0, 0, 0,]]

『検索リスト』
[[1, 0],
[0, 0]]

『前提』
並びが一致する箇所は、必ず一箇所だけ存在する
並びが完全に一致していない場合は、無視

■検索リストと並びが一致しているところは、下記太字部分

[[0, 0, 0,],
[1, 0, 1,],
[0, 1, 0,],
[0, 0, 0,]]

出力したいインデックスは、
2 1
※2 1というのは、並びが完全に一致している先頭部分「検索リストの最初（１）」が見つかった、対象リストのインデックスです。

よろしくお願いします。

Answer 1

numpyを使えば楽にできる方法があるかもと思いつつ、ただのpythonで書いてみました。

lst = [[0, 0, 0],
       [1, 0, 1],
       [0, 1, 0],
       [0, 0, 0]]
target = [[1,0],
          [0,0]]

def search_target(lst, target):
    h = len(lst)  # 高さ
    w = len(lst[0])  # 幅
    t_h = len(target)  # 検索対象の高さ
    t_w = len(target[0])  # 検索対象の幅
    loop_h = h - t_h + 1  # 縦方向のループ回数（検索対象の高さに被らないところまで＋検索対象と同じ高さ分）
    loop_w = w - t_w + 1  # 横方向のループ回数
    
    def get_partial(i, j):  # indexから部分リストを返す
        return [row[j:j+t_w] for row in lst[i:i+t_h]]  # lst[i:i+t_h]でまず垂直方向に取り出す。取り出した各行を今度はrow[j:j+t_w]で水平方向に切り出す

    # ループ
    for i in range(loop_h):  
        for j in range(loop_w):
            if target == get_partial(i,j):  # 部分リストと検索対象が一致したら位置を返す
                return (i, j)
    return None

print(search_target(lst, target))  # => (2, 1)

Comments

[zeitaku_fire]

ありがとうございます。
自分の作成したコードでは、60行近くになってしまって、かつ正しい動きをしていませんでした。。
プログラミング能力が高い人だと、こんなに短いコードで要件を満たせるのですね。

ご教授頂いたコードについて、どこで何を処理しているのか、まだ完璧に理解できてませんが、こちらのコードを参考に、自分もこんなコードが書けるよう精進していきたいと思います。
お忙しい中ありがとうございました。

[hayataka2049]

適当にコメントを入れてみたので、参考にしてください