python - 从生成器获取特定的生成器数据

Question

让我们从代码开始

class MyClass:
   def __init__(self):
      self.elemplusone = None
      self.elemplustwo = None
      self.data = self.generate_data()
      
   def generate_data(self):
      for elem in range(10):
         yield elem+1, elem+2

我需要获取生成器的第一个和第二个元素。 现在，我在创建 object 之后在 class 之外调用它：

a_generator = MyClass()
c = next(a_generator.data)
elemplusone = c[0]
elemplustwo = c[1]

但我需要在 class 中指定它们（作为单独的生成器），我无法创建两个生成器方法。

谢谢

Answer 1

我也不太明白你到底是什么意思。 但这对你有帮助吗？

class MyClass:
   def __init__(self):
      self.data = self.generate_data()
      self.update_elements()

   def update_elements(self):
      self.elemplusone, self.elemplustwo = [x for x in next(self.data)]

   def generate_data(self):
      for elem in range(10):
          print("Yielded")
          yield elem + 1, elem + 2

a_generator = MyClass()

a_generator.elemplusone为 1， a_generator.elemplustwo为 2。

现在您可以调用a_generator.update_elements()再次生成您的元素并在您的生成器中继续。 如果这对您有帮助，请告诉我。 祝你好运！