Python多线程池的使用方法

发布时间:2024-08-30 17:24:32

阅读量:81

Python的concurrent.futures库提供了ThreadPoolExecutor类，它是一个高效的线程池实现

导入所需模块：

import concurrent.futures

定义一个函数，该函数将在线程池中运行：

def my_function(x):
    # 在这里执行你的任务
    return x * x

创建一个ThreadPoolExecutor实例，并指定线程池中的最大线程数。然后，使用submit()方法将任务添加到线程池中：

with concurrent.futures.ThreadPoolExecutor(max_workers=5) as executor:
    # 提交任务并将Future对象存储在列表中
    futures = [executor.submit(my_function, i) for i in range(10)]

使用as_completed()方法迭代已完成的任务，并获取结果：

for future in concurrent.futures.as_completed(futures):
    result = future.result()
    print(result)

下面是一个完整的示例：

import concurrent.futures

def my_function(x):
    return x * x

with concurrent.futures.ThreadPoolExecutor(max_workers=5) as executor:
    futures = [executor.submit(my_function, i) for i in range(10)]

for future in concurrent.futures.as_completed(futures):
    result = future.result()
    print(result)