Python技术分享—生成器

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接: https://blog.csdn.net/TT789789789/article/details/94556450

生成器

Python技术分享—生成器

有时候,序列或集合内的元素的个数非常巨大,如果全制造出来并放入内存,对计算机的压力是非常大的。比如,假设需要获取一个10**20次方如此巨大的数据序列,把每一个数都生成出来,并放在一个内存的列表内,这是粗暴的方式,有如此大的内存么?如果元素可以按照某种算法推算出来,需要就计算到哪个,就可以在循环的过程中不断推算出后续的元素,而不必创建完整的元素集合,从而节省大量的空间。在Python中,这种一边循环一边计算出元素的机制,称为生成器:generator。

生成生成器:

g = (x * x for x in range(1, 4))
g
<generator object <genexpr> at 0x1022ef630>

可以通过next()函数获得generator的下一个返回值,这点和迭代器非常相似:

next(g)
1
next(g)
4
next(g)
9
next(g)
Traceback (most recent call last):
 File "<pyshell#14>", line 1, in <module>
 next(g)
StopIteration
------------------------------------------------
但更多情况下,我们使用for循环。
for i in g:
 print(i)

除了使用生成器推导式,我们还可以使用yield关键字。

def createNums():
 print("----func start------")
 a,b = 0,1
 for i in range(5):
 # print(b)
 print("--1--")
 yield b
 print("--2--")
 a,b = b,a+b # a,b = 1, 1 a,b = 1,2
 print("--3--")
 print("----func end------")
g= createNums()
next(g) # 如果想得到yield的值,可以打印next(g)

在 Python中,使用yield返回的函数会变成一个生成器(generator)。 在调用生成器的过程中,每次遇到yield时函数会暂停并保存当前所有的运行信息,返回yield的值。并在下一次执行next()方法时从当前位置继续运行。

# 斐波那契函数
def fibonacci(n): 
 a = 0
 b = 1
 counter = 0
 while True:
 if counter > n:
 return
 yield a # yield让该函数变成一个生成器
 a, b = b, a + b
 counter += 1
fib = fibonacci(10) # fib是一个生成器
print(type(fib))
for i in fib:
 print(i, end=" ")

生成器是可以循环的,相比next来说,for循环更友好

a = createNums()
这两种取值方式是一样的!!!
a.__next__()
next(a)
for i in a:
 print(i)

send

def test():
 i = 0
 while i<5:
 temp = yield i
 print(temp)
 i+=1
t = test()
next(t)
next(t)
t.send("juran")
next(t)
--------------------------------------------
t = test()
t.send("juran")
Traceback (most recent call last):
 File "/Users/binbin/Desktop/Python/demo.py", line 179, in <module>
 t.send("juran")
TypeError: can't send non-None value to a just-started generator
如何解决这个错误?
> next(t)
 t.send("juran")

> send(None)
生成器的应用

实现多任务

def test1():
 while True:
 print("--1--")
 yield None

def test2():
 while True:
 print("--2--")
 yield None
t1 = test1()
t2 = test2()
while True:
 next(t1)
 next(t2)

(写在最后,由于以后每天晚上九点半之后会更新Python基础的知识点,记得来看哦!)

此文来源于微博和今日头条:逻二妞,转载请注明出处,谢谢

推荐阅读

改变世界到语言—python,看完才知道从事python凭什么高薪!(2019年最新三大方向资源)

这本书教你读懂大脑中的神经元,并一个月内掌握Python基础,经典

猜你喜欢

转载自blog.csdn.net/TT789789789/article/details/94556450