生成器

最新推荐文章于 2024-07-16 08:24:58 发布

greybeard

最新推荐文章于 2024-07-16 08:24:58 发布

阅读量83

点赞数

分类专栏： python 文章标签： yield python

本文链接：https://blog.csdn.net/greybeard/article/details/84094855

版权

python 专栏收录该内容

85 篇文章 0 订阅

订阅专栏

任何包含yield语句的函数称为生成器。
它的行为和普通的函数有很大的差别，它不是像return那样返回值，而是每次产生
多个值。每次产生一个值(使用yield语句),函数就会被冻结:即函数停在那点等待
被激活。函数被激活后就从停止的那点开始执行。

例子:


def flatten(nested):
    for sublist in nested:
        for element in sublist:
            yield element

nested = [[1,2], [3, 4], [5]]

print list(flatten(nested))

递归生成器:


# recursion yield example
def flatten2(nested):
    try:
        # Don't iteration of the string similar objects
        try: nested + ''
        except TypeError: pass
        else: raise TypeError
        for sublist in nested:
            for element in flatten2(sublist):
                yield element
    except TypeError:
        yield nested

print list(flatten2([[[1],2], 3, 4, [5, [6, 7]], 8]))

print list(flatten2(['foo', ['bar', ['baz']]]))