Python线程：如何让程序跑得更快？

小鹿( ﹡ˆoˆ﹡ )

于 2024-09-27 09:37:56 发布

阅读量274

点赞数 7

分类专栏： Python 文章标签： python 前端数据库 Python

本文链接：https://blog.csdn.net/qq_44771627/article/details/142585558

版权

Python 专栏收录该内容

67 篇文章 1 订阅

订阅专栏

引言

线程（Thread）是操作系统能够进行运算调度的最小单位，它被包含在进程之中，是进程中的实际运作单位。一个进程可以拥有多个线程，这些线程共享进程的数据空间，并能并发执行，从而提高程序的效率。在Python中使用线程，可以让我们在编写网络爬虫、多任务处理等应用时更加游刃有余。接下来，我们将从最基础的概念出发，一步步深入了解线程的创建与启动。

基础语法介绍

在Python中，threading模块是处理线程的主要方式。通过这个模块，我们可以轻松地创建线程对象，并启动它们来执行特定的任务。

创建线程

创建线程的基本步骤如下：

导入threading模块。
定义一个继承自threading.Thread类的新类，并重写其run()方法。run()方法将定义该线程要执行的具体操作。
创建上述类的实例，即创建了一个新的线程对象。
调用线程对象的start()方法来启动线程。

启动线程

启动线程后，线程将开始执行其run()方法中定义的代码。如果想要等待线程完成，可以调用线程对象的join()方法。

基础实例

假设我们需要同时执行两个耗时的操作，比如下载两个不同的网页，就可以通过创建两个线程来实现并发下载。

import threading
import requests

class DownloadThread(threading.Thread):
    def __init__(self, url):
        super().__init__()
        self.url = url
    
    def run(self):
        response = requests.get(self.url)
        print(f"Downloaded {self.url}, status code: {response.status_code}")

urls = ["http://example.com", "http://example.org"]
threads = []

for url in urls:
    thread = DownloadThread(url)
    threads.append(thread)
    thread.start()

# 等待所有线程完成
for thread in threads:
    thread.join()

这段代码首先定义了一个DownloadThread类，该类继承自threading.Thread，并在其run()方法中实现了下载功能。然后，我们创建了两个这样的线程实例，并分别对两个URL发起了请求。

进阶实例

当涉及到大量并发请求或者复杂的线程间通信时，简单的线程管理可能不足以满足需求。这时，可以考虑使用线程池或队列来更好地组织和控制线程的行为。

例如，使用concurrent.futures.ThreadPoolExecutor来管理一个固定大小的线程池，可以有效地控制并发数量，并简化线程的管理和结果的收集过程。

from concurrent.futures import ThreadPoolExecutor

def download_url(url):
    response = requests.get(url)
    return f"Downloaded {url}, status code: {response.status_code}"

with ThreadPoolExecutor(max_workers=5) as executor:
    futures = [executor.submit(download_url, url) for url in urls]
    
    # 收集所有线程的结果
    for future in futures:
        print(future.result())