少说话多写代码之Python学习052——类的成员(实现生成器)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/yysyangyangyangshan/article/details/84679310

前面我们知道类中定义__iter__可以实现迭代器,我们再看看另一个概念,叫生成器。生成器和迭代器是Python中近几年引入的显著地两个特性。
生成器是用普通的函数语法定义的迭代器。
又这样一个数组:sequen=[[1,2],[3,4],[5]]。我们要按顺序打印出来,

def flatten(seq):
    for list in seq:
        for item in list:
            yield  item

代码中,yield是一个新的语句,函数中包含了yield语句,则该函数称之为生成器。使用yield每次产生一个值,函数就被冻结,函数停在这个点直到被重新唤醒后从该点继续执行。
看看调用

sequen=[[1,2],[3,4],[5]]
for item in flatten(sequen):
    print(item)

输出

1
2
3
4
5

也可以这样使用

print(list(flatten(sequen)))

输出

[1, 2, 3, 4, 5]

工程文件下载:https://download.csdn.net/download/yysyangyangyangshan/10821883

猜你喜欢

转载自blog.csdn.net/yysyangyangyangshan/article/details/84679310