如何仅从生成器中选择一项？

Question

我有一个类似下面的生成器函数：

def myfunct():
  ...
  yield result

调用此函数的常用方法是：

for r in myfunct():
  dostuff(r)

我的问题是，有什么方法可以随时从生成器中获取一个元素吗？例如，我想做类似的事情：

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

Answer 1

使用创建一个生成器

g = myfunct()

每当您想要一件物品时，请使用

next(g)

（或g.next()在Python 2.5或更低版本中）。

如果发电机退出，它将升高StopIteration。您可以根据需要捕获此异常，也可以将default参数使用到next()：

next(g, default_value)