首先可以简单理解一下yield关键字,包含了关键字yield的函数,可以被视为一个生成器,这个生成器有更丰富的功能,并且这个生成器是随用随生成的,下面的例子说明了这点:
def get_next():
for i in range(10):
yield i
if __name__ == "__main__":
g = get_next()
# 输出结果完全一致,都是0-10的数字
print("yield生成结果:")
print([x for x in g])
print("传统生成器结果:")
print([x for x in range(10)])
两次print生成的结果完全一样,说明get_next函数就是一个生成器。
按照网上其他人说法,假设生成100万个数字,传统生成器需要实现把这一百万个数字准备好,而yield则是随用随取;假设一下,你现在只用了前5个数据,后面的直接忽略了,那么传统生成器生成100万个数字就有浪费了。这一点可以看其他人的验证。
因此,可以认为yield有两个用处,一是提供功能更丰富的的生成器,二是提供更节约资源的生成器。
下面来看一下实际效果。
参照别人的代码,进行了部分修改,主要对inner和outer两个值进行了比较,以此来帮助理解yield与传统生成器的区别。
其中结合业务逻辑,可以假设outer是真正业务中需要的值,这个值由call(i)函数生成,分别是0,2,4,6,8
关于yield,它就是一个类似于return的关键词。类似于生成器,调用__next__或者send()方法时,yield会立即返回一个值并停止执行,当下一次调用__next__或者send()方法的时候,会从停止执行的位置继续执行
最需要额外注意的是,如果yield函数与其他变量结合(例如代码中的inner)并被外界调用,如果是通过__next__调用的,那么inner会为空,如果是通过send()调用的,inner会被设置为相应的值;二者的区别在于返回之后,如何填补原来的空位。
接下来,通过inner变量,可以看出next和send的区别,可以认为send包含了next。
在get_next()函数中打印“inner”的值,这个值在step1中不会被打印,
在step2和step3中,“inner”的值为none,这是因为,在"yield call(i)"时,直接返回了call(i),并且将none赋给了“inner”;在外界是通过__next__调用的.
在step4中,因为调用了send方法,因此在在"yield call(i)"时,不仅仅返回了call(i),还将将外部send发送进来的值赋给了“inner”,
在step5中,send()函数发送进来的值可以是任意值,只要满足业务要求,也就是说这个send进来的值将作用于inner;
代码如下:
def call(i):
return i * 2
def get_next():
for i in range(100):
inner = yield call(i)
print("inner----", inner)
if __name__ == "__main__":
g = get_next()
print("**********step 1**********")
print("outer----", g.__next__())
print("**********step 2**********")
print("outer----", g.__next__())
print("**********step 3**********")
value_to_send = g.__next__()
print("outer----", value_to_send)
print("**********step 4**********")
value_to_send = g.send(value_to_send)
print("outer----", value_to_send)
print("**********step 5**********")
value_to_send = g.send("达到最大步骤5,需要提前终止")
print("outer----", value_to_send)
# 运行结果:
# **********step 1**********
# outer---- 0
# **********step 2**********
# inner---- None
# outer---- 2
# **********step 3**********
# inner---- None
# outer---- 4
# **********step 4**********
# inner---- 4
# outer---- 6
当不考虑inner变量,仅仅考虑外部的outer变量时,情况就变得简单了;注意,此时仍然能够通过send()函数继续进行数字生成,并且效果和__next__是相似的:
def call(i):
return i * 2
def get_next():
for i in range(100):
yield call(i)
if __name__ == "__main__":
g = get_next()
print("**********step 1**********")
print("outer----", g.__next__())
print("**********step 2**********")
print("outer----", g.__next__())
print("**********step 3**********")
print("outer----",g.send("输入不会影响yield"))
# 输出结果
# **********step 1**********
# outer---- 0
# **********step 2**********
# outer---- 2
# **********step 3**********
# outer---- 4