再帰関数でyieldを利用したい

はじめに

再帰関数でyieldを利用したい場合はyield fromを利用します。

例題

以下のような仮想的なディレクトリ構造をリストで表現し、それを再帰的に探索するジェネレーターを作ってみましょう。

tree: list[tuple[str, str]] = [
    ("root", "folder1"),
    ("root", "folder2"),
    ("root", "file4.txt"),
    ("folder1", "file1.txt"),
    ("folder1", "file2.txt"),
    ("folder2", "folder3"),
    ("folder3", "file3.txt"),
]

yield from を使った再帰ジェネレーター

def walk(
    tree: list[tuple[str, str]], stack: list[str], current: str
) -> Iterator[list[str]]:
    if current.endswith(".txt"):
        yield stack

    for parent, child in tree:
        if parent == current:
            stack.append(child)
            yield from walk(tree, stack, child)
            stack.pop()

この walk 関数は、ディレクトリ構造を再帰的にたどりながら、ファイルのパスを1つずつジェネレートします。

実行例

for path in walk(tree, ["root"], "root"):
    print(path)

出力例:

['root', 'folder1', 'file1.txt']
['root', 'folder1', 'file2.txt']
['root', 'folder2', 'folder3', 'file3.txt']
['root', 'file4.txt']

関連リンク

• Python公式ドキュメント: yield from

実行環境

• Python 3.13.7