tqdm python使用总结

一 概述

参考:https://blog.csdn.net/winter2121/article/details/111356587

总结:tqdm()只是给遍历过程添加了进度条,传入的可以是可迭代对象,也可以是不可迭代对象(需要额外设置,成为手动更新)。这两种运行模式具体见下面链接:
https://zhuanlan.zhihu.com/p/163613814

二 使用示例

# 可迭代对象,自动更新
dic = ['a', 'b', 'c', 'd', 'e']
pbar = tqdm(dic)
for i in pbar:
    pbar.set_description('Processing '+i)
    time.sleep(0.2)  
 
 Processing e: 100%|██████████| 5/5 [00:01<00:00,  4.69it/s]
# 不可迭代对象,手动更新
from tqdm import tqdm
import time

with tqdm(range(100), desc='Test') as tbar:
    for i in tbar:
        tbar.set_postfix(loss=i/100, x=i)
        tbar.update()  # 手动更新,默认参数n=1,每update一次,进度+n
        time.sleep(0.2)

在这里插入图片描述

# 不可迭代对象,手动更新
from tqdm import tqdm
import time

with tqdm(range(100), desc='Test') as tbar:
    for i in tbar:
        tbar.set_postfix({'loss': '{0:1.5f}'.format(i/100), "x": '{0:1.5f}'.format(i)})
        tbar.update()  # 手动更新,默认参数n=1,每update一次,进度+n
        time.sleep(0.2)

在这里插入图片描述

# 不可迭代对象,手动更新
from tqdm import tqdm
import time

total = 200 #总迭代次数
loss = total

with tqdm(total=200, desc='Test') as pbar:
    pbar.set_description('Processing:') #  tqdm()中的desc参数和pbar.set_description()是一回事,后者优先。
    for i in range(20):
        loss -= 1
        pbar.set_postfix({'loss': '{0:1.5f}'.format(i/100), "x": '{0:1.5f}'.format(i)})  # 输入一个字典,显示实验指标
        # pbar.set_postfix(loss=i/100, x=i)
        pbar.update(10)  # 手动更新,默认参数n=1,每update一次,进度+n
        time.sleep(0.2)

在这里插入图片描述
总结:tqdm.set_postfix()传入参数的两种方式:

# 1 传入元组
tbar.set_postfix(loss=i/100, x=i)

# 2 传入字典
tbar.set_postfix({'loss': '{0:1.5f}'.format(i/100), "x": '{0:1.5f}'.format(i)})

# 两种方法的结果一样
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐