时间戳

首页 > 解决方案 > 将 tqdm 与 concurrent.futures 一起使用？

python - 将 tqdm 与 concurrent.futures 一起使用？

问题描述

我有一个多线程函数，我想要一个状态栏来使用tqdm. 有没有一种简单的方法来显示状态栏ThreadPoolExecutor？令我困惑的是并行化部分。

import concurrent.futures

def f(x):
    return f**2

my_iter = range(1000000)

def run(f,my_iter):
    with concurrent.futures.ThreadPoolExecutor() as executor:
        function = list(executor.map(f, my_iter))
    return results

run(f, my_iter) # wrap tqdr around this function?

标签： pythonconcurrent.futurestqdm

解决方案

您可以tqdm围绕executor以下内容跟踪进度：

list(tqdm(executor.map(f, iter), total=len(iter))

这是您的示例：

import time  
import concurrent.futures
from tqdm import tqdm

def f(x):
    time.sleep(0.001)  # to visualize the progress
    return x**2

def run(f, my_iter):
    with concurrent.futures.ThreadPoolExecutor() as executor:
        results = list(tqdm(executor.map(f, my_iter), total=len(my_iter)))
    return results

my_iter = range(100000)
run(f, my_iter)

结果是这样的：

16%|██▏           | 15707/100000 [00:00<00:02, 31312.54it/s]

推荐阅读