【Python】生成器和递归

l=[1, 2, 3, 4, 5, 6]

如果l求和，毫无疑问可以使用递归，比如可以这样：

 def sum(l):
     res = 0
     for i in l:
         if not isinstance(i, list):
             res+=i
         else:
             res+=sum(i)
     return res

如果我们想把l中的所有子列表裂解开，这种方法行不行。。。？比如可以这样

方法1：

 def play(l):
     res = []
     for i in l:
         if isinstance(i, list):
             t=list(play(i))
             for j in t:
                 res.append(j)
         else:
             res.append(i)
     return res

如果使用生成器的话：

方法2

 def play2(l):
     try:
         for i in l:
             for num in play2(i):
                 yield num
     except TypeError:
         yield l

可以看到方法2看起来简单了很多，但是方2都有一个问题，如果有一个列表是这样的

l2=['abs',['de','fg',['hi','jk']]]

那么方法2还可行吗？

这是方法2的输出结果：

方法2进入了无限递归。。。因为字符串不会引起Type Error

那该怎么办？

 def play3(l):
     try:
         try: l+''
         except TypeError: pass
         else: raise TypeError
         for i in l:
             for s in play3(i):
                 yield s
     except TypeError:
         yield l

用一个简单的l+''检查是否为字符串，如果l是字符串的话，进入else引起Type Error，然后yield l

如果l不是字符串，则什么都不干，进入下面的for循环。

很有意思的用法