生成器面试题1

def demo():  # 生成器函数
    for i in range(4):
        yield i

g = demo()  # 得到一个生成器，demo中的代码没有被执行

g1 = (i for i in g)     # ()所以是生成器表达式，得到了一个生成器g1，生成器表达式的代码一句也没有被执行
g2 = (i for i in g1)    # 生成器表达式，得到了一个生成器g2，生成器表达式的代码没有被执行

print(list(g1))     # [0, 1, 2, 3]      # 此时使用数据类型强制转换的方式从生成器g1中取值，所以g1对应的生成器表达式开始执行，全部执行完毕，g1生成器没有东西了
print(list(g2))     # []    # 此时取值，g2对应的表达式开始执行，表达式中遍历g1生成器，但是g1生成器值已经取完了，所以这里为[]

# -------------------------------------------

def add(n, i):
    return n + i

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

g = test()

for n in [1, 10]:
    g = (add(n, i) for i in g)  # 拆开看

print(list(g))  # [20, 21, 22, 23]


# n = 1
#g = (add(n, i) for i in test())

# n = 2
# g = (add(n, i) for i in (add(n, i) for i in test()))

#list(g)