Python tqdm显示代码任务进度

用户9925864

发布于 2022-07-27 08:46:11

1.9K0

发布于 2022-07-27 08:46:11

文章被收录于专栏：算法工程师的学习日志

有时候在使用Python处理比较耗时操作的时候，为了便于观察处理进度，这时候就需要通过进度条将处理情况进行可视化展示，以便我们能够及时了解情况。

tqdm就能非常完美的支持和解决这些问题，可以实时输出处理进度而且占用的CPU资源非常少，支持windows、Linux、mac等系统，支持循环处理、多进程、递归处理等进度展示。

安装tqdm

github地址：https://github.com/tqdm/tqdm

想要安装tqdm也是非常简单的，通过pip就可以安装，而且不需要安装其他的依赖库。

pip install tqdm -i https://pypi.tuna.tsinghua.edu.cn/simple

迭代对象处理

对于可以迭代的对象都可以使用下面这种方式，来实现可视化进度，非常方便

from tqdm import tqdm
import time

for i in tqdm(range(100)):
    time.sleep(0.1)
    pass

在使用tqdm的时候，可以将tqdm(range(100))替换为trange(100)代码如下

from tqdm import tqdm,trange
import time

for i in trange(100):
  time.sleep(0.1)
  pass

通过tqdm提供的set_description方法可以实时查看每次处理的数据

from tqdm import tqdm
import time

str = tqdm(["a", "b", "c"])
for c in str:
    time.sleep(1)
    str.set_description("Processing %s" % c)

手动设置处理的进度

通过update方法可以控制每次进度条更新的进度

from tqdm import tqdm
import time

# total参数设置进度条的总长度
with tqdm(total=100) as pbar:
    for i in range(100):
        time.sleep(0.05)
        # 每次更新进度条的长度
        pbar.update(1)

自定义进度条显示信息

通过set_description和set_postfix方法设置进度条显示信息

from tqdm import trange
from random import random, randint
import time

with trange(100) as t:
    for i in t:
        # 设置进度条左边显示的信息
        t.set_description("GEN %i" % i)
        # 设置进度条右边显示的信息
        t.set_postfix(loss=random(), gen=randint(1, 999), str="h", lst=[0, 1, 2])
        time.sleep(0.1)

多层循环进度条

通过tqdm也可以很简单的实现嵌套循环进度条的展示

在pycharm中执行以上代码的时候，会出现进度条位置错乱，目前官方并没有给出好的解决方案，这是由于pycharm不支持某些字符导致的，不过可以将上面的代码保存为脚本然后在命令行中执行，效果如下

from tqdm import tqdm
import time

for i in tqdm(range(20), ascii=True, desc="1st loop"):
    for j in tqdm(range(10), ascii=True, desc="2nd loop"):
        time.sleep(0.01)