Python快速入门(十四)协程:迭代器、生成器、yield、greenlet、gevent
- 1.迭代器
- 2.生成器
- 3.通过异常来判断生成器结束
- 4.通过send来启动生成器
- 5.使用yield完成多任务
- 6.使用greenlet来完成多任务
- 7.使用gevent完成多任务
前言:主要讲解了迭代器、生成器、使用yield、greenlet、gevent来完成多任务。
1.迭代器
迭代是访问集合元素的一种方式。迭代器是一个可以记住遍历的位置的对象。迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退。
自己实现可迭代的对象,代码:
import time
class Classmate(object):
def __init__(self):
self.names = list()
self.current_num = 0
def add(self, name):
self.names.append(name)
def __iter__(self):
"""如果想要一个对象称为一个 可以迭代的对象,即可以使用for,那么必须实现__iter__方法"""
return self # 调用iter(xxobj)的时候 只要__iter__方法返回一个 迭代器即可,至于是自己 还是 别的对象都可以的, 但是要保证是一个迭代器(即实现了 __iter__ __next__方法)
def __next__(self):
if self.current_num < len(self.names):
ret = self.names[self.current_num]
self.current_num += 1
return ret
else:
raise StopIteration
classmate = Classmate()
classmate.add("老王")
classmate.add("王二")
classmate.add("张三")
for name in classmate:
print(name)
time.sleep(1)
输出:
2.生成器
利用迭代器,我们可以在每次迭代获取数据(通过next()方法)时按照特定的规律进行生成。但是我们在实现一个迭代器时,关于当前迭代到的状态需要我们自己记录,进而才能根据当前状态生成下一个数据。为了达到记录当前状态,并配合next()函数进行迭代使用,我们可以采用更简便的语法,即生成器(generator)。生成器是一类特殊的迭代器。
生成器demo:
def create_num(all_num):
print("----1---")
a, b = 0, 1
current_num = 0
while current_num < all_num:
print("----2---")
yield a
print("----3---")
a, b = b, a+b
current_num += 1
print("----4---")
obj = create_num(10)
ret = next(obj)
print(ret)
ret = next(obj)
print(ret)
obj2 = create_num(2)
ret = next(obj2)
print(ret)
输出:
3.通过异常来判断生成器结束
代码:
def create_num(all_num):
a, b = 0,1
current_num = 0
while current_num < all_num:
yield a
a, b = b, a+b
current_num += 1
return "ok..."
obj2 = create_num(50)
while True:
try:
ret = next(obj2)
print(ret)
except Exception as ret:
print(ret.value)
break
运行:
4.通过send来启动生成器
代码:
def create_num(all_num):
a, b = 0, 1
current_num = 0
while current_num < all_num:
ret = yield a
print(ret)
a, b = b, a+b
current_num += 1
obj = create_num(10)
ret = next(obj)
print(ret)
ret = obj.send("hahah")
print(ret)
运行:
5.使用yield完成多任务
import time
def task_1():
while True:
print("------1-----")
time.sleep(0.1)
yield
def task_2():
while True:
print("------2-----")
time.sleep(0.1)
yield
def main():
t1 = task_1()
t2 = task_2()
while True:
next(t1)
next(t2)
if __name__ == '__main__':
main()
运行截图:
6.使用greenlet来完成多任务
代码:
from greenlet import greenlet
import time
def test1():
while True:
print("---A--")
gr2.switch()
time.sleep(0.5)
def test2():
while True:
print("---B--")
gr1.switch()
time.sleep(0.5)
gr1 = greenlet(test1)
gr2 = greenlet(test2)
#切换到gr1中运行
gr1.switch()
7.使用gevent完成多任务
代码:
import gevent
import time
def f1(n):
for i in range(n):
print(gevent.getcurrent(), i)
gevent.sleep(0.5)
def f2(n):
for i in range(n):
print(gevent.getcurrent(), i)
gevent.sleep(0.5)
def f3(n):
for i in range(n):
print(gevent.getcurrent(), i)
gevent.sleep(0.5)
print("------1-----")
g1 = gevent.spawn(f1, 5)
print("------2-----")
g2 = gevent.spawn(f2, 5)
print("------3-----")
g3 = gevent.spawn(f3, 5)
print("------4-----")
g1.join()
g2.join()
g3.join()
结束!!!