qq_44687077的博客

原创 tqdm和imap合用——多进程显示进度条

最近在用multiprocessing库的pool.map函数开启多个进程处理数据时，由于数据量较大，程序计算过程没有提示，不知道运行到哪一步了，也不知道计算速度怎么样。因此想着用tqdm来显示进度。但是，pool.map函数只有当所有进程都算完之后才能返回计算结果，即最开始进度0%，然后等到所有的结果都计算完后，进度直接100%，看不到中间的结果。因此，考虑用imap代替map。imap用法和map函数一样，外面再套上tqdm函数和list函数即可。