歩留まりを使用した再帰

再帰とyieldステートメントを組み合わせる方法はありますか？たとえば、（再帰を使用する）無限数ジェネレータは次のようになります。

def infinity(start):
    yield start
    # recursion here ...

>>> it = infinity(1)
>>> next(it)
1
>>> next(it)
2

私は試した：

def infinity(start):
    yield start
    infinity(start + 1)

そして

def infinity(start):
    yield start
    yield infinity(start + 1)

しかし、それらのどれも私が望むことをしませんでした、最初のものはそれが降伏した後に停止startし、2番目のものは降伏しstart、次にジェネレーターそしてそれから停止しました。

注： whileループを使用してこれを実行できることを知っています。

def infinity(start):
    while True:
        yield start
        start += 1

これを再帰的に実行できるかどうかを知りたいだけです。

はい、あなたはこれを行うことができます：

def infinity(start):
    yield start
    for x in infinity(start + 1):
        yield x

ただし、最大再帰深度に達すると、これはエラーになります。

Python 3.3以降、次を使用できるようになります

def infinity(start):
    yield start
    yield from infinity(start + 1)

ジェネレーター関数をループしたり、ループしたりせずに再帰的に呼び出す場合yield fromは、関数本体を実際に実行したり、何も生成したりせずに、新しいジェネレーターを作成するだけです。

詳細については、PEP380を参照してください。

場合によっては、ジェネレーターに再帰ではなくスタックを使用する方が望ましい場合があります。スタックとwhileループを使用して再帰メソッドを書き直すことができるはずです。

コールバックを使用し、スタックロジックを使用して書き換えることができる再帰メソッドの例を次に示します。

def traverse_tree(callback):
    # Get the root node from somewhere.
    root = get_root_node()
    def recurse(node):
        callback(node)
        for child in node.get('children', []):
            recurse(child)
    recurse(root)

上記のメソッドは、各ノードにchildren子ノードを含む可能性のある配列があるノードツリーをトラバースします。各ノードが検出されると、コールバックが発行され、現在のノードがノードに渡されます。

この方法はこのように使用でき、各ノードにいくつかのプロパティを出力します。

def callback(node):
    print(node['id'])
traverse_tree(callback)

代わりにスタックを使用し、トラバーサルメソッドをジェネレーターとして記述します

# A stack-based alternative to the traverse_tree method above.
def iternodes():
    stack = [get_root_node()]
    while stack:
        node = stack.pop()
        yield node
        for child in reversed(node.get('children', [])):
            stack.append(child)

（最初と同じトラバーサル順序が必要な場合は、スタックに追加された最初の子が最後にポップされる子になるため、子の順序を逆にする必要があることに注意してください。）

これで、traverse_tree上記と同じ動作を得ることができますが、ジェネレーターを使用します。

for node in iternodes():
    print(node['id'])

これは万能の解決策ではありませんが、一部のジェネレーターでは、再帰の代わりにスタック処理を使用すると、良い結果が得られる場合があります。

def lprint(a):
    if isinstance(a, list):
        for i in a:
            yield from lprint(i)
    else:
        yield a

b = [[1, [2, 3], 4], [5, 6, [7, 8, [9]]]]
for i in lprint(b):
    print(i)

したがって、基本的には、関数を再帰的に呼び出す必要がある場所にforループを追加する必要があります。これはPython2.7に適用されます。