ジェネレーターをループする方法

ジェネレーターをどのようにループできますか？私はこのように考えました：

gen = function_that_returns_a_generator(param1, param2)
if gen: # in case the generator is null
    while True:
        try:
            print gen.next()
        except StopIteration:
            break

もっとパイソン的な方法はありますか？

単に

for x in gen:
    # whatever

トリックを行います。if gen常にを返すことに注意してくださいTrue。

for item in function_that_returns_a_generator(param1, param2):
    print item

関数から返されるものがあるかどうかを確認するためのテストについて心配する必要はありません。返されるものがないかのように、ループに入ることはありません。

副作用のみを気にするためにジェネレーターの出力が必要ない場合は、次のワンライナーを使用できます。

for _ in gen: pass

あなたは単にそれをループすることができます：

>>> gen = (i for i in range(1, 4))
>>> for i in gen: print i
1
2
3

ただし、ループできるのは1回だけであることに注意してください。次回ジェネレーターは空になります：

>>> for i in gen: print i
>>> 

他の反復​​可能なものと同じように扱ってください。

for val in function_that_returns_a_generator(p1, p2):
    print val

これif gen:は常にTrueになるため、誤ったテストであることに注意してください

ジェネレーターを手動で移動する場合（つまり、各ループを手動で操作する場合）、次のようにすることができます。

    from pdb import set_trace

    for x in gen:
        set_trace()
        #do whatever you want with x at the command prompt
        #use pdb commands to step through each loop of the generator e.g., >>c #continue   

他の答えは、複雑なシナリオに適しています。単にアイテムをリストにストリーミングしたい場合：

x = list(generator)

（または、ジェネレーターをトリガーして何かを実行したいだけの場合は、単にlist(generator)。

単純な前処理には、リスト内包表記を使用します。

x = [tup[0] for tup in generator]

または、単純な関数を実行する場合：

# didn't assign to variable b/c we don't care about what the print() function returns
[print(x) for x in gen]