python多进程打印进度条

最新推荐文章于 2024-11-26 23:25:27 发布

原创最新推荐文章于 2024-11-26 23:25:27 发布 · 5.3k 阅读

9 ·

CC 4.0 BY-SA版权

Python 编程专栏收录该内容

133 篇文章

订阅专栏

本文介绍了一种使用Python的multiprocessing模块进行并行处理的方法，并结合tqdm库实现任务进度的可视化显示。通过一个具体示例，展示了如何创建进程池、分配任务并实时更新进度条。

import time
from tqdm import tqdm
import multiprocessing as mp


def pickle_process(_class, *args):
    return _class.proc_func(*args)


class OP():
    def __init__(self):
        self.length = 64

    def proc_func(self):
        time.sleep(0.1)

    def flow(self):
    	# ------------- 配置好进度条 ------------- 
        pbar = tqdm(total=self.length)
        pbar.set_description(' Flow ')
        update = lambda *args: pbar.update()
        # --------------------------------------
        pool = mp.Pool(16)
        for _ in range(self.length):
            pool.apply_async(pickle_process, args=(self, ), callback=update)	# 通过callback来更新进度条
        pool.close()
        pool.join()


if __name__ == '__main__':

    start_time = time.time()
    op = OP()
    op.flow()
    print(time.time() - start_time)