python教程（九）之特性（6.递归生成器）

#试想一下，如果我想把[[1,2],[3,4],5]变为[1,2,3,4,5],需要做什么
#大致的办法你应该可以很快想到，利用循环遍历，如果是列表就拆开
#那如果[[1,[2],3],[4,5]]甚至更多层，用循环还能处理的了吗
#递归会是你最好的选择，利用我们之前学的生成器，我们可以更好地做到

def flatten(arg) :
    try :
        for sublist in arg :
            for element in flatten(sublist) :
                yield element
    except TypeError :
        yield arg            
a=[[1,[2],3],[4,5]]
print(list(flatten(a)))     
#[1, 2, 3, 4, 5]

#如果一个函数定义中包含yield关键字，那么这个函数就不再是一个普通函数，而是一个generator
#generator和函数的执行流程不一样，函数是顺序执行，遇到return语句或者最后一行函数语句就返回
#变成generator的函数，在每次调用next()的时候执行，遇到yield语句返回，再次执行时从上次返回的yield语句处继续执行

#也许你还不是很明白，我这里解释一下
#上面的代码中没有用到next，也就不涉及到从上次返回的yield语句处继续执行这个特点
#生成器返回的是一个算法，只有你去访问它时才会给你一个可读的结果

for i in flatten(a) :
    print(i)
# 1
# 2
# 3
# 4
# 5
print(flatten(a))#<generator object flatten at 0x0000000001F9C620>

#这里出一道题，下一次会公布答案
# 期待输出:
# [1]
# [1, 1]
# [1, 2, 1]
# [1, 3, 3, 1]
# [1, 4, 6, 4, 1]
# [1, 5, 10, 10, 5, 1]
# [1, 6, 15, 20, 15, 6, 1]
# [1, 7, 21, 35, 35, 21, 7, 1]
# [1, 8, 28, 56, 70, 56, 28, 8, 1]
# [1, 9, 36, 84, 126, 126, 84, 36, 9, 1]
# 这是个杨辉三角，不知道的可以查一下，使用递归生成器，生成这样的结果