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

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

（或在 Python 2.5 或更低版本中使用 g.next()）创建生成器

g = myfunct()

使用每次您想要一个项目时，使用

next(g)

。

如果生成器退出，它将引发 StopIteration。如果需要，您可以捕获此异常，或者使用 next() 的 default 参数：

next(g, default_value)

Answer 2

要仅选取生成器的一个元素，请在 for 语句中使用 break 或 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 异常处理或额外的默认值比较。

仅当您想在生成器结束时执行一些特殊操作时，才需要 for 语句部分的 else: 。

关于 next() / .next() 的注意事项：

在 Python3 中，.next() 方法已重命名为 .__next__() 有充分的理由：它被认为是低级别的（PEP 3114）。在 Python 2.6 之前，内置函数 next() 不存在。甚至讨论了将 next() 移至 operator 模块（这本来是明智的），因为它的需求很少且内置名称的膨胀值得怀疑。

在没有默认值的情况下使用 next() 仍然是非常低级的做法 - 在正常应用程序代码中像晴天霹雳一样公开抛出神秘的 StopIteration 。并且将 next() 与默认哨兵一起使用 - 这最好应该是直接在 builtins 中使用 next() 的唯一选项 - 是有限的，而且经常给出了奇怪的非Python逻辑/可读性的理由。

底线：使用 next() 应该很少 - 就像使用 operator 模块的函数一样。使用 for x in iterator 、 islice 、 list(iterator) 和其他无缝接受迭代器的函数是在应用程序级别使用迭代器的自然方式- 而且总是有可能的。 next() 是低级的，一个额外的概念，不明显 - 正如该线程的问题所示。例如，在 for 中使用 break 是常规做法。

Answer 3

Generator 是一个生成迭代器的函数。因此，一旦有了迭代器实例，请使用 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)

Answer 4

您可以使用解构来选择特定项目，例如：

>>> first, *middle, last = range(10)
>>> first
0
>>> middle
[1, 2, 3, 4, 5, 6, 7, 8]
>>> last
9

请注意，这将消耗您的生成器，因此虽然可读性很高，但它的效率低于 next() 之类的东西，并且对无限生成器来说是毁灭性的：

>>> first, *rest = itertools.count()

Answer 5

我不相信有一种方便的方法可以从生成器中检索任意值。生成器将提供 next() 方法来遍历自身，但不会立即生成完整序列以节省内存。这就是生成器和列表之间的功能差异。

Answer 6

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

确保捕获最后一个元素被获取后抛出的异常

Answer 7

对于那些浏览这些答案以获取 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

Answer 8

这是从迭代器获取单个对象的简短示例。如果您知道您要查找的对象的属性或属性，则会将其提供给您。

obj = next((obj for obj in iterator() if obj.prop == match_prop), None)
if obj is not None:
    pass

Answer 9

我相信唯一的方法是从迭代器获取列表，然后从该列表中获取所需的元素。

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