简单聊聊Python协程

最新推荐文章于 2024-07-15 08:45:25 发布

罗罗攀

最新推荐文章于 2024-07-15 08:45:25 发布

阅读量924

点赞数

文章标签： python java 编程语言多线程机器学习

本文链接：https://blog.csdn.net/luoluopan/article/details/120662661

版权

本文介绍了Python协程的基本概念和异步编程的重要性。通过实例展示了如何使用asyncio模块来优化代码执行效率，将原本需要13秒的刷牙烧水过程缩短至8秒。协程通过在IO等待期间切换执行其他任务，提升了代码执行效率，尤其适用于有大量IO操作的场景。

摘要由CSDN通过智能技术生成

前言

今天我们来聊聊Python协程，当Python学习到一定的深度，当你需要对代码进行优化提速时，就避不开异步编程，尤其是现在很多优秀的第三方库都实现了异步编程，这使得我们不得不学习。

本人也是最近开始学习这个概念，本次分享我们就来简单聊聊Python协程的概念和最基本的语法。

什么是协程

我们先来看看协程的概念。

协程（Coroutine），也可以被称为微线程，是一种用户态内的上下文切换技术。

概念是不是很不好理解，没关系，我们引入一个小学就学过的数学思考题。小明早上起床需要刷牙和烧水，刷牙需要5分钟，烧水需要8分钟，问小明总共需要几分钟完成这两件事情。

答案是8分钟，这你应该是知道的吧。其实这就是协程，现在我再来解释下，刷牙是一个函数，烧水时一个函数，那协程就是通过一个线程实现代码块相互切换执行。

Python实现协程

我们首先来看看不用协程怎么来写。

import time


def brush_teeth():
    print('开始刷牙')
    time.sleep(5)
    print('刷牙完成')


def boil_water():
    print('开始烧水')
    time.sleep(8)
    print('烧水完成')


def main():
    start = time.time()
    brush_teeth()
    boil_water()
    end = time.time()
    print('花费时间{}s'.format(end-start))


main()

开始刷牙
刷牙完成
开始烧水
烧水完成
花费时间13.01053786277771s

看到没，需要13s，那我们来看看协程怎么写。

import asyncio
import time


async def brush_teeth():
    print('开始刷牙')
    await asyncio.sleep(5)
    print('刷牙完成')


async def boil_water():
    print('开始烧水')
    await asyncio.sleep(8)
    print('烧水完成')


async def main():
    start = time.time()
    tasks = [
        asyncio.create_task(boil_water()),
        asyncio.create_task(brush_teeth())
    ]
    await asyncio.wait(tasks)
    end = time.time()
    print('花费时间{}s'.format(end-start))


asyncio.run(main())

开始烧水
开始刷牙
刷牙完成
烧水完成
花费时间8.004523992538452s

用了协程后，程序就只需要8s了。所以协程很有用，协程一般应用在有IO操作的程序中，因为协程可以利用IO等待的时间去执行一些其他的代码，从而提升代码执行效率。