Как пройти через генератор

Как можно пройти через генератор? Я думал об этом так:

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

Но имейте в виду, что цикл можно выполнить только один раз. В следующий раз генератор будет пуст:

>>> 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]