Добавить переменную / метод члена к генератору 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