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

         此段代码看起来更加简洁明了、通俗易懂,将递归生成器的原理展现的淋漓尽致。

你可能感兴趣的