前言
在进行比较耗时任务时,通常需要查看程序完成了多少任务,比较读取1亿行的数据,通过进度条能可视化展示当前读取了到多少行。又比如进行网络训练时,训练1万个epoch,也可以通过进度条查看训练多少个epoch了,以及检查程序是否运行正常。
一、tqdm是什么?
示例:python 写的一种进度条可视化工具
from tqdm import tqdm
text = str()
for item in tqdm(["a","b","c","d","e","f"]):
time.sleep(1)
text += item
print(text)
结果
列表总共有6个元素,每个元素相当于一个任务,完成一个任务需要1秒,进度条可动态显示完成了多少任务
二、我最常使用的两种风格
1.把迭代对象封装到tqdm
代码如下(示例):
for i in tqdm(range(100)):
time.sleep(0.25)
结果
2.打开tqdm窗口
代码如下(示例):
pbar = tqdm(total=100)
for i in range(10):
time.sleep(0.25)
pbar.update(2)
pbar.close()
这种方式能合理设置任务更新节奏,update的参数表示每次任务更新步数,上述代码执行完更新了20%。
总结
以上是tqdm这个工具最常用两种方式,大家也可以尝试其它用法。