Question

我有一个生成器，我想知道我是否可以使用它而不必担心StopIteration，我想在没有 for item in generator 的情况下使用它。我想将它与while语句（或其他构造）一起使用。我怎么能这样做？

Answer 1

内置功能

next（迭代器[，默认]）
通过调用其__next__()方法从迭代器中检索下一个项。如果给定default，则在迭代器耗尽时返回，否则引发StopIteration。

在Python 2.5及更早版本中：

raiseStopIteration = object()
def next(iterator, default=raiseStopIteration):
    if not hasattr(iterator, 'next'):
       raise TypeError("not an iterator")
    try:
       return iterator.next()
    except StopIteration:
        if default is raiseStopIteration:
           raise
        else:
           return default

Answer 2

另一个选择是一次读取所有生成器值：

>>> alist = list(agenerator)

示例：

>>> def f():
...   yield 'a'
...
>>> a = list(f())
>>> a[0]
'a'
>>> len(a)
1

Answer 3

用它来包裹你的发电机：

class GeneratorWrap(object):

      def __init__(self, generator):
          self.generator = generator

      def __iter__(self):
          return self

      def next(self):
          for o in self.generator:
              return o
          raise StopIteration # If you don't care about the iterator protocol, remove this line and the __iter__ method.

像这样使用：

def example_generator():
    for i in [1,2,3,4,5]:
        yield i

gen = GeneratorWrap(example_generator())
print gen.next()  # prints 1
print gen.next()  # prints 2

更新：请使用下面的答案，因为它比这个好得多。

有没有另一种方法在python生成器上调用next？

3 个答案: