您的位置:

用Python的tqdm和enumerate实现循环进度条

一、什么是tqdm和enumberate?

tqdm是一个Python进度条库,可以在循环时自动创建进度条,让任务执行更加可视化。简单使用tqdm的方法如下所示:

from tqdm import tqdm

for i in tqdm(range(1000000)):
    pass

执行过程中会在命令行自动创建一个进度条,如下所示:

100%|██████████| 1000000/1000000 [00:01<00:00, 961178.26it/s]

enumberate是Python内置的函数,可以同时获取循环中元素的值和索引。简单使用enumberate的方法如下所示:

a = ['a', 'b', 'c']

for index, value in enumberate(a):
    print(index, value)

输出结果:

0 a
1 b
2 c

二、使用tqdm和enumberate实现循环进度条

使用tqdm和enumberate可以非常方便地实现循环进度条,代码如下所示:

import time
from tqdm import tqdm

a = ['a', 'b', 'c']

for index, value in tqdm(enumerate(a)):
    time.sleep(1)

执行过程中会在命令行自动创建一个进度条,如下所示:

33%|███▎      | 1/3 [00:01<00:02,  1.02s/it]
67%|██████▋   | 2/3 [00:02<00:01,  1.02s/it]
100%|██████████| 3/3 [00:03<00:00,  1.02s/it]

可以看到,每次循环都会自动更新进度条,显示当前循环的进度。

三、自定义进度条的样式

tqdm默认使用的是ascii字符作为进度条的样式,如下所示:

import time
from tqdm import tqdm

a = ['a', 'b', 'c']

for index, value in tqdm(enumerate(a)):
    time.sleep(1)

输出结果:

33%|███▎      | 1/3 [00:01<00:03,  1.02s/it]
67%|██████▋   | 2/3 [00:02<00:01,  1.03s/it]
100%|██████████| 3/3 [00:03<00:00,  1.03s/it]

如果想要改变进度条的样式,可以通过tqdm的参数进行自定义。tqdm提供了五种样式供选择,分别是"ascii", "ascii-ext", "unicode", "unicode-l1", "ncurses"。使用方法如下所示:

import time
from tqdm import tqdm

a = ['a', 'b', 'c']

# 通过tqdm参数设置进度条样式
for index, value in tqdm(enumerate(a), bar_format="{l_bar}%s{bar}%s{r_bar}" % (Fore.BLUE, Fore.RESET)):
    time.sleep(1)

其中,bar_format参数可以用来自定义进度条的样式。可以在进度条前后添加任何需要的字符,例如可以添加颜色,如上面代码中的"{l_bar}%s{bar}%s{r_bar}" % (Fore.BLUE, Fore.RESET),使用了tqdm内置的颜色库,将进度条设置为蓝色。

四、结语

tqdm和enumberate是两个非常实用的Python库和函数,在Python编程中经常使用。结合使用tqdm和enumberate可以非常方便地创建循环进度条,使任务执行更加可视化,也能够更直观地了解任务的执行情况。

完整代码如下所示:

import time
from tqdm import tqdm
from colorama import Fore

a = ['a', 'b', 'c']

# 通过tqdm参数设置进度条样式
for index, value in tqdm(enumerate(a), bar_format="{l_bar}%s{bar}%s{r_bar}" % (Fore.BLUE, Fore.RESET)):
    time.sleep(1)