yield就是为了节省内存的。
只有你调用next(),代码才会开始执行,最终:
1.返回yield后面的值给调用方(也就是next())
2.并暂停到yeild的地方(记住是暂停,千万不要当成某些资料说的return来理解)
注意上面的描述哈,一定是调用next(),生成器(先不要管这个名词)代码才会开始运行!
抛出问题:
给你一些数字,你将他们全部+1
于是你轻松写下代码:
print([i+1 for i in range(10)])
问题来了,如果是一亿个数呢?你可以试试把10改为100000000,看看你电脑卡不卡
我尝试了一下:
反正我在这里卡了半天!
那有没有解决方法呢?
我们将上述的中括号换成括号即可!(generator)
这玩意就是生成器,你不调用它,他是不会运行的。我们可以用next()来调用,以获取下一个值。
换成我们平时所见的代码:
def f123():
for i in range(10000000000000000000000000000000):
print(f'before: {i}')
yield i+1 #代码会返回i+1给调用方(也就是next()),然后暂停到这里
#print([x+1 for x in range(10000000000000000000000000000000)])
# g就是生成器啦!
g = f123()
# for循环,隐式调用next(),你可以自行换成g.__next__()
for item in g:
print(f'after add:{item}')
#休眠可以让我们观察控制台结果
time.sleep(1)
注意python3是g.__next__(),2好像不是。
当然这个结果,你只能遍历一次。是不可能像普通list那样,来回遍历的。