python之递归生成器[yield用法]
作者:互联网
在Magnus Lie Hetland所写的畅销书籍《Python基础教程》中讲到了递归生成器的用法。其中,作者举了一个例子,假设有一个列表nested:
nested = [[[1],2],3,4,[5,[6,7]]]
一、晦涩的程序
如果我们想要依次打印列表中的数字,需要用到递归生成器,作者Magnus Lie Hetland给出的程序如下:
def flatten(nested):
try:
for sublist in nested:
for element in flatten(sublist):
yield element
except TypeError:
yield nested
输出得到的结果:
list(flatten(nested))
毫无疑问,上述程序得到的结果是正确。但是,作者所写的程序让人不敢恭维。我刚开始看的时候是一头雾水,等看了很多遍后,才理解了程序想要表达的意思。可以这么说,作者的这段代码并不适合用来教学,因为这段代码晦涩难懂。
二、易懂的程序
其实,递归生成器非常简单,主要分为两部分:基线条件和递归条件。Magnus Lie Hetland直接将try..except...语句的来作为基线条件和递归条件,实在有点让人难以理解。
我们以易懂通俗的方式来解释递归生成器的用法,并且依次打印列表nested中的数字。
1、基线条件
基线条件很简单,如果迭代的元素是数字,那么直接yield该元素。
2、迭代条件
如果迭代的元素不是数字,则继续继续迭代。
def flatten(nested):
for sublist in nested:
#基线条件
if isinstance(sublist,int):
yield sublist
#递归条件
else:
for element in flatten(sublist):
yield element
此段代码看起来更加简洁明了、通俗易懂,将递归生成器的原理展现的淋漓尽致。
标签:递归,python,生成器,sublist,yield,nested,flatten 来源: https://blog.csdn.net/hanyunkaka/article/details/120676836