一、什么是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)