版权声明:本文为博主原创文章,未经博主允许不得转载。 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