Fügen Sie einem Python-Generator eine Elementvariable/Methode hinzu?

Question

Kann ich einem Python-Generator eine Elementvariable/-methode hinzufügen?

Ich möchte etwas in den folgenden Zeilen, so dass ich kann "Peek" an Membervariable 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 

Ja, ich weiß, dass ich i und j jedes Mal zurückgeben kann. Aber ich will das nicht tun. Ich möchte einen lokalen Generator sehen.

Answer 1

Sie könnten ein Objekt erstellen und manipulieren die __iter__ Schnittstelle:

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 

Drucke:

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

Ich denke, es ist hässlich, aber es sollte tun, was Sie wollen. Ich würde eher zurückkehren i und j jedes Mal, wenn :-)

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