tqdm用法总结

文章目录

  • 前言
  • 一、tqdm是什么?
  • 二、我最常使用的两种风格
    • 1.把迭代对象封装到tqdm
    • 2.打开tqdm窗口
  • 总结


前言

在进行比较耗时任务时,通常需要查看程序完成了多少任务,比较读取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这个工具最常用两种方式,大家也可以尝试其它用法。

你可能感兴趣的:(python,python,机器学习,深度学习,自然语言处理)