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)

输出:

python转换为脉冲信号的方法_生成器

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)

输出:

python转换为脉冲信号的方法_迭代器_02

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

运行:

python转换为脉冲信号的方法_多任务_03

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)

运行:

python转换为脉冲信号的方法_Python协程_04

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()

运行截图:

python转换为脉冲信号的方法_迭代器_05

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()

结束!!!