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

我有一个发电机,我想知道我是否可以使用它而不必担心StopIteration,我想在没有发电机的for项目的情况下使用它.我想将它与while语句(或其他构造)一起使用.我怎么能这样做？

解决方法:

用它来包裹你的发电机：

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,language-features
来源： https://codeday.me/bug/20190713/1452772.html