Python：用tqdm模块绘制进度条

最新推荐文章于 2024-01-23 08:00:00 发布

转载最新推荐文章于 2024-01-23 08:00:00 发布 · 358 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：https://www.cnblogs.com/orion-orion/p/16033773.html

文章标签：

#it

在计算密集型迭代计算中，我们常常需要知道当前的迭代轮次，最传统的方法就是打印当前迭代计数器的轮数。那有没有更好的方法呢？我们可以使用tqdm模块（非py内置，需要用conda/pip单独安装）来在控制台绘制进度条，这样更加直观。

最简单的一种方法是直接做为迭代器使用：

from tqdm import tqdmfrom time import sleepn_rounds = 10for i in tqdm(range(n_rounds)):    sleep(1)

可以看到，循环每迭代一次，进度条就会+10%（我们这里按enter键换行显示）：

 0%|                                                     河南红色教育培训 www.bjganxun.cn                | 0/10 [00:00<?, ?it/s] 10%|█████▉                                                     | 1/10 [00:01<00:09,  1.00s/it] 20%|███████████▊                                               | 2/10 [00:02<00:08,  1.00s/it] 30%|█████████████████▋                                         | 3/10 [00:03<00:07,  1.00s/it] 40%|███████████████████████▌                                   | 4/10 [00:04<00:06,  1.00s/it] 50%|█████████████████████████████▌                             | 5/10 [00:05<00:05,  1.00s/it] 60%|███████████████████████████████████▍                       | 6/10 [00:06<00:04,  1.00s/it] 70%|█████████████████████████████████████████▎                 | 7/10 [00:07<00:03,  1.00s/it] 80%|███████████████████████████████████████████████▏           | 8/10 [00:08<00:02,  1.00s/it] 90%|█████████████████████████████████████████████████████      | 9/10 [00:09<00:01,  1.00s/it]100%|██████████████████████████████████████████████████████████| 10/10 [00:10<00:00,  1.00s/it]

还有一种稍复杂的方法，就是自定义进度条计数器的更新。比如下面我们规定迭代到一个奇数则进度条计数器+1，迭代到5个奇数进度条计数器满。

from tqdm import tqdmfrom time import sleepn_rounds = 10n_odd = 5pbar = tqdm(total=n_odd)for i in range(n_rounds):    sleep(1)    if i % 2 == 1:        pbar.update(1)

可以看到，每迭代到一个奇数，进度条就会+20%（我们这里按enter键换行显示）。

 0%|                                                                    | 0/5 [00:00<?, ?it/s] 20%|████████████                                                | 1/5 [00:02<00:08,  2.01s/it] 40%|████████████████████████                                    | 2/5 [00:04<00:06,  2.01s/it] 60%|████████████████████████████████████                        | 3/5 [00:06<00:04,  2.01s/it] 80%|████████████████████████████████████████████████            | 4/5 [00:08<00:02,  2.00s/it]100%|████████████████████████████████████████████████████████████| 5/5 [00:10<00:00,  2.16s/it]