python的一段代码解读

Question

def add(s, x):
    return s + x

def gen():
    for  i in range(4):
        yield i

base = gen()
for n in [1, 10]:
    base = (add(i,  n) for i in base)

print list(base)

上面的代码为啥输出[20,21,22,23],哪位大神给详细解读一下

Answer 1

n等于1时，执行完base = (add(i, n) for i in base)后base是i+n for i in [0,1,2,3],接下来n=10，base变为i+n for i in (i+n for i in [0,1,2,3]), 此时n等于10，再去计算base, base一直是个生成器。其中print list(base) 是对生成器进行迭代，此时n是按照10来计算。

Answer 2

那个for循环里
第一次是这样的
base = (i+n for i in base)
第二次是这样的
base = (i+n+n for i in base)

Answer 3

哈哈，我前段时间也碰到这个问题了！ 理解了好长时间，终于弄明白了，记录在博客了～～可以看看，应该有帮助:python迭代器与生成器小结。