У мене функція генератора:

def myfunct():
  ...
  yield result

Звичайним способом викликати цю функцію було б:

for r in myfunct():
  dostuff(r)

Моє запитання, чи є спосіб отримати лише один елемент з генератора, коли мені подобається? Наприклад, я хотів би зробити щось на кшталт:

while True:
  ...
  if something:
      my_element = pick_just_one_element(myfunct())
      dostuff(my_element)
  ...

Створіть генератор за допомогою

g = myfunct()

Кожен раз, коли ви хочете предмет, використовуйте

next(g)

(або g.next()в Python 2.5 або нижче).

Якщо генератор вийде, він підніметься StopIteration. Ви можете зловити цей виняток, якщо це необхідно, або скористайтеся defaultаргументом, щоб next():

next(g, default_value)

Для вибору лише одного елемента генератора використовуйте breakу forоператорі, абоlist(itertools.islice(gen, 1))

За вашим прикладом (буквально) ви можете зробити щось на кшталт:

while True:
  ...
  if something:
      for my_element in myfunct():
          dostuff(my_element)
          break
      else:
          do_generator_empty()

Якщо ви хочете " отримати лише один елемент з [колись створеного] генератора, коли мені це подобається " (я вважаю, що 50% - це початковий намір і найпоширеніший намір), тоді:

gen = myfunct()
while True:
  ...
  if something:
      for my_element in gen:
          dostuff(my_element)
          break
      else:
          do_generator_empty()

Цим способом generator.next()можна уникнути явного використання , а обробка в кінці вводу не вимагає (критичного) StopIterationоброблення виключень або додаткових порівнянь за замовчуванням.

else:З forрозділу заяви необхідно тільки , якщо ви хочете зробити що - щось особливе в разі відслужили генератора.

Примітка про next()/ .next():

У Python3 .next()метод було перейменовано з .__next__()поважної причини: його вважали низьким рівнем (PEP 3114). До Python 2.6 вбудованої функції next()не існувало. І навіть обговорювалося перехід next()до operatorмодуля (що було б розумно) через рідкісну потребу та сумнівну інфляцію побудованих назв.

Використання next()без замовчування все ще є дуже низьким рівнем практики - відкрито викидати криптовалюту StopIterationяк болт у звичайний код програми. А використання next()дозорного режиму за замовчуванням - що найкраще має бути єдиним варіантом для next()безпосередньо в builtins- обмежене і часто дає підстави для непарної непітонічної логіки / читабельності.

Підсумок: Використання next () має бути дуже рідкісним - як використання функцій operatorмодуля. Використовуючи for x in iterator, islice, list(iterator)та інші функції , приймаючи итератор легко природний спосіб використання ітераторів на рівні додатків - і цілком можливо завжди. next()є низькорівневим, додатковою концепцією, непомітною - як показує питання цієї теми. Хоча, наприклад, використання breakв forзвичайному.

Я не вірю, що існує зручний спосіб отримати довільне значення з генератора. Генератор забезпечить наступний () метод для проходження себе, але повна послідовність не створюється негайно для збереження пам'яті. Ось функціональна різниця між генератором і списком.

Для тих, хто сканує ці відповіді, повний робочий приклад для Python3 ... ну ось вам:

def numgen():
    x = 1000
    while True:
        x += 1
        yield x

nums = numgen() # because it must be the _same_ generator

for n in range(3):
    numnext = next(nums)
    print(numnext)

Це виводи:

1001
1002
1003

Генератор - це функція, яка виробляє ітератор. Тому, як тільки у вас є екземпляр ітератора, використовуйте next () для отримання наступного елемента з ітератора. Як приклад, використовуйте функцію next () для отримання першого елемента, а пізніше for inдля обробки решти елементів:

# create new instance of iterator by calling a generator function
items = generator_function()

# fetch and print first item
first = next(items)
print('first item:', first)

# process remaining items:
for item in items:
    print('next item:', item)

generator = myfunct()
while True:
   my_element = generator.next()

переконайтеся, що вилучити виняток, кинутий після взяття останнього елемента

Я вважаю, що єдиний спосіб - отримати список із ітератора, а потім отримати потрібний елемент із цього списку.

l = list(myfunct())
l[4]