yield关键字一直困扰了我很久,一直也没有弄明白,现在将暂时理解的yield记录如下,供参考:
关键词:可迭代对象,生成器,迭代器
一、可迭代对象:
可迭代对象:可迭代对象是一个泛称,只要可以用for...in...处理的对象都可以称为可迭代对象。包括:列表(list)、元组(tuple)、字典(dict)、字符串(str)、文件(file)及迭代器和生成器。
二、迭代器:
迭代器:迭代器是一个对象,实现__iter__方法和__next__方法(只实现了__iter__方法的对象是可迭代的,并实现了__next__方法的对象是迭代器),迭代器通过调用__next__()方法获取下一个值;
自定义迭代器(实现__iter__()方法和__next__()方法)
class M(object):
def __init__(self,max):
self.max=max
self.n=0
def __iter__(self):
return self
def __next__(self):
if self.n<self.max:
self.n=self.n+1
return self.n
else:
raise StopIteration #一定要加上这一句,否则是无限循环了
m=M(5)
for i in m:
print(i)
输出结果:
1
2
3
4
5
三、生成器:
生成器:生成器是一个函数,内部实现了yield关键字的函数都可称为生成器。
生成器是特殊的迭代器。它不像list那样,将所有的数据都存储在内存中,那样占用极大的内存空间,而生成器每一次调用只返回一个数值,因此效率更高。
生成器的优点:效率高,速度快,不占内存,优选使用生成器而不是list之类的可迭代对象
生成器生成有两种个方式:
1.(x*2 for x in range(5))即可返回一个生成器
2.实现了yield关键字的函数。带有yield的函数就不再是一个普通的函数,而变成了一个生成器可以迭代了。
yield理解:
1.yield相当于普通函数中的return语句,yield语句右边即是其包含函数(生成器)要返回的值。
2.每次迭代,运行完yield语句后暂停(即将yield右侧值返回),yield下一行语句等到下一次迭代再运行。
3.生成器可以迭代的原因是内部实现了一个__next()__方法,每次迭代,运行一次__next__()方法,直到捕获到异常为止。
4.迭代三种方式:for..in..循环;调用__next__()方法;调用send(msg)方法
代码示例(来源于网络,https://www.jianshu.com/p/d09778f4e055):
# encoding:UTF-8
def yield_test(n):
for i in range(n):
yield call(i)
print("i=", i)
# 做一些其它的事情
print("do something.")
print("end.")
def call(i):
return i * 2
# 使用for循环
for i in yield_test(5):
print(i, ",")
#或者使用__next__()方法
for i in range(5):
yield_test(5).__next__()
输出结果:
>>>
0 ,
i= 0
2 ,
i= 1
4 ,
i= 2
6 ,
i= 3
8 ,
i= 4
do something.
end.
>>>
yield生成器的__next__()方法与send(msg)方法
参考文章:
2.python的迭代器为什么一定要实现__iter__方法?
由于生成器是惰性运行,即每次迭代只运行一次。我们定义一个迭代器:
def h():
print('wen chuan')
m=yield 5#将yield 5当作一个整体,赋值给m
print(m)
n=yield 12
print('together')
if __name__=='__main__':
g=h()#生成一个生成器
print(type(g))#验证g的类型,打印值为<class 'generator'>
g.__next__()#迭代一次
print(g.send('haha'))
打印结果:
<class 'generator'>
wen chuan
haha
12
执行过程: 执行g.__next__(),程序运行到yield 5,将5返回,然后暂停。执行g.send('haha'),将yield 5当作整体,将‘haha'赋值给这个整体,并赋值给m,打印m值,再次运行到yield 12,将12返回。
因此,send()函数除了具有__next__()一模一样的迭代作用外,还能够给jield x整体赋值。