将成员变量/方法添加到Python生成器？

Question

我可以将成员变量/方法添加到Python生成器吗？

我想大意如下的东西，让我可以 “窥视” 的成员变量j：

def foo(): 
    for i in range(10): 
     self.j = 10 - i 
     yield i 

gen = foo() 
for k in gen: 
    print gen.j 
    print k 

是的，我知道我可以每次都返回i和j。但我不想这样做。我想窥视发电机内的一个本地。

Answer 1

你可以创建一个对象，并操纵__iter__接口：

class Foo(object): 
    def __init__(self): 
     self.j = None 
    def __iter__(self): 
     for i in range(10): 
      self.j = 10 - i 
      yield i 

my_generator = Foo() 

for k in my_generator: 
    print 'j is',my_generator.j 
    print 'k is',k 

打印：

j is 10 
k is 0 
j is 9 
k is 1 
j is 8 
k is 2 
j is 7 
k is 3 
j is 6 
k is 4 
j is 5 
k is 5 
j is 4 
k is 6 
j is 3 
k is 7 
j is 2 
k is 8 
j is 1 
k is 9 

Answer 2

我觉得它很丑，但它应该做你想做的。我宁愿回到I和J每一次，虽然:-)

class Foo(object): 
    def foo(self): 
     for i in range(10): 
      self.j = 10 - i 
      yield i 

genKlass = Foo() 
gen = genKlass.foo() 
for k in gen: 
    print genKlass.j 
    print k