test6.py

# 练习生成式，生成式就是将列表生成式的中括号改成小括号即可


g = (x for x in range(1,20))
print(g)  # 这是个generator对象
print(next(g))  # 不断使用next来进行获取下一个数，当获取到最后一个数时会抛出StopIteration错误
# 可见这种方法对于列表相当大时，就可以节约内存，同时他也可以用for循环来获取元素
for x in g:
    print(x)


# 使用yield来定义一个生成器的函数
def odd():
    print('yield 1')
    # yield就是一个关键字，用来生成生成器
    # 每执行一个next函数时，就会运行到一个yield处，把该值返回
    yield 1
    print('yield 2')
    yield 2
    print('yield 3')
    yield 3


o = odd()
# 他是生成器，所以可以通过next来获取数
print(next(o))
print(next(o))
print(next(o))
# next(o)


# 使用yield来编写斐波那契数列
def fib(n):
    a, b = 0, 1
    while n > 0:
        # 第一和第二个数都是1
        yield b
        a, b = b, a+b
        n -=1
    return 'done'


num = fib(6)
# 因为是生成器，所以可以通过for循环来获取元素
for x in num:
    print(x)


# 作业：有一个杨辉三角，把每一行看做一个list，试写一个generator，不断输出下一行的list：
def triangles(n):
    l = [1]
    # 记录第几行，同时知道有多少个元素
    num = 1
    while num <= n:
        # 使用切片是避免两个变量指向同一个列表，从而影响了下面操作
        L = l[:]
        yield L
        num += 1
        if len(l) > 1:
            for d in range(1, num-1):
                l[d] = L[d] + L[d-1]
        l[0] = 1
        l.append(1)


t = triangles(10)
for x in t:
    print(x)