python库01—进度条库tqdm

Top Secret

已于 2023-07-12 10:19:17 修改

阅读量320

点赞数

CC 4.0 BY-SA版权

分类专栏： python数据分析与数据处理文章标签： python 前端 javascript

于 2023-07-12 09:20:39 首次发布

原文链接：https://zhuanlan.zhihu.com/p/163613814

python数据分析与数据处理专栏收录该内容

24 篇文章

订阅专栏

文章介绍了Python的tqdm模块，用于创建进度条。它支持基于迭代对象和手动更新两种运行模式，并提供了多种参数来自定义进度条，包括描述文字、总项目数、保留进度条选项等。示例展示了如何使用tqdm显示进度条以及改变颜色和单位。

tqdm模块是python进度条库, 主要分为两种运行模式：

1.基于迭代对象运行: tqdm(iterator)

import time
from tqdm import tqdm

# trange(i)是tqdm(range(i))的一种简便写法
for i in range(100):
    time.sleep(0.05)

for i in tqdm(range(100), desc='Processing_zard'):
    time.sleep(0.05)

dic = ['a', 'b', 'c', 'd', 'e']
pbar = tqdm(dic)
for i in pbar:
    pbar.set_description('Processing_'+i)
    time.sleep(0.2)

结果：

C:\Users\ZARD\anaconda3\envs\pytorch_gpu\python.exe C:/Users/ZARD/PycharmProjects/pythonProject/代码手术台/tqdm_demo.py
Processing_zard: 100%|██████████| 100/100 [00:06<00:00, 16.06it/s]
Processing_e: 100%|██████████| 5/5 [00:01<00:00,  4.93it/s]

2.手动进行更新

import time
from tqdm import tqdm

with tqdm(total=200) as pbar:
    pbar.set_description('Processing_zard:')
    # total表示总的项目, 循环的次数20*10(每次更新数目) = 200(total)
    for i in range(20):
        # 进行动作, 这里是过0.1s
        time.sleep(0.1)
        # 进行进度更新, 这里设置10个
        pbar.update(10)

结果：

C:\Users\ZARD\anaconda3\envs\pytorch_gpu\python.exe C:/Users/ZARD/PycharmProjects/pythonProject/代码手术台/tqdm_demo.py
Processing_zard:: 100%|██████████| 200/200 [00:02<00:00, 91.43it/s]

3.tqdm模块参数说明

class tqdm(object):
  """
  Decorate an iterable object, returning an iterator which acts exactly
  like the original iterable, but prints a dynamically updating
  progressbar every time a value is requested.
  """

  def __init__(self, iterable=None, desc=None, total=None, leave=False,
               file=sys.stderr, ncols=None, mininterval=0.1,
               maxinterval=10.0, miniters=None, ascii=None,
               disable=False, unit='it', unit_scale=False,
               dynamic_ncols=False, smoothing=0.3, nested=False,
               bar_format=None, initial=0, gui=False):

iterable: 可迭代的对象, 在手动更新时不需要进行设置
desc: 字符串, 左边进度条描述文字
total: 总的项目数
leave: bool值, 迭代完成后是否保留进度条
file: 输出指向位置, 默认是终端, 一般不需要设置
ncols: 调整进度条宽度, 默认是根据环境自动调节长度, 如果设置为0, 就没有进度条, 只有输出的信息
unit: 描述处理项目的文字, 默认是'it', 例如: 100 it/s, 处理照片的话设置为'img' ,则为 100 img/s
unit_scale: 自动根据国际标准进行项目处理速度单位的换算, 例如 100000 it/s >> 100k it/s

实操：

# 实操
import time
from tqdm import tqdm

def action():
    time.sleep(0.5)
with tqdm(total=1000,desc='文字说明',leave=True,ncols=100,unit='B',unit_scale=True) as pbar:
    for i in range(10):
        # 休眠0.5秒
        action()
        # 更新休眠进度
        pbar.update(100)

结果：

C:\Users\ZARD\anaconda3\envs\pytorch_gpu\python.exe C:/Users/ZARD/PycharmProjects/pythonProject/代码手术台/tqdm_demo.py
文字说明: 100%|██████████████████████████████████████████████████| 1.00k/1.00k [00:05<00:00, 198B/s]

4.tqdm调整进度条的颜色

from tqdm import tqdm
import time

data = range(100)
data_bar = tqdm(data, colour='#C0FF20')                    # 实例化tqdm对象，包含颜色
for x in data_bar:
    time.sleep(0.1) # 进度条每0.1s前进一次，总时间为1000*0.1=100s
    data_bar.set_description(f'文字说明，当前数值为:{x}')     # 实时更新tqdm的表头

如：