Python 学习

Appearance

进度条 tqdm ​

tqdm 允许任何可迭代的对象使用进度条的形式展示，它可以协助监测程序运行的进度，只需要封装可迭代对象即可。

安装 ​

使用 pip3 包管理工具直接进行安装，命令如下：

bash
pip3 install tqdm  # 安装 tqdm 扩展包

pip3 show tqdm # 检测 tqdm 模块是否可用

常用参数 ​

tqdm 构造函数的参数众多，可以在这里查看更多，下面列举常用参数：

  • iterable: 最常用的参数，表示使用这个迭代对象来初始化 tqdm 对象，如果手动更新进度条的话该参数可以为 None
  • desc: 进度条的描述信息
  • total: 进度条总格子数量
  • leave: 布尔值, 迭代完成后是否保留进度条
  • file: 输出指向位置, 默认是终端, 一般不需要设置
  • ncols: 调整进度条宽度, 默认是根据环境自动调节长度, 如果设置为 0, 就没有进度条, 只有输出的信息
  • miniters: int或可选。迭代过程中进度显示的最小更新间隔。
  • unit: str或可选。定义每个迭代的单元。默认为 it，即每个迭代，在下载或解压时，设为"B"，代表每个“字节”。
  • unit_scale: bool或者int或者float或者可选。默认为 False，如果设置为 1 或者 True，会自动根据国际单位制进行转换 (kilo, mega, etc.) 。比如，在下载进度条的例子中，如果为 False，数据大小是按照字节显示，设为 True 之后转换为Kb、Mb。

自动控制进度 ​

  • 将 tqdm() 应用在任意迭代器

    python
    from tqdm import tqdm
import time

for i in tqdm(range(1000)):
    time.sleep(0.01)

# 100%|███████████████████| 1000/1000 [00:11<00:00, 86.71it/s]

  • 使用 trange 代替 tqdm(range(xxx))

    python
    from tqdm import trange
import time

for i in trange(1000):
    time.sleep(0.01)

# 100%|███████████████████| 1000/1000 [00:11<00:00, 86.51it/s]

  • 添加描述和单元

    python
    from tqdm import tqdm
import time

for i in tqdm(range(1000), desc="My tqdm:", unit="B"):
    time.sleep(0.01)

# My tqdm:: 100%|███████████████████| 1000/1000 [00:11<00:00, 86.95B/s]

    上述进度条中会添加描述

  • 动态描述

    使用 set_description() 方法在进度条前面添加描述性内容，动态设置描述可以让其看起来更加的人性化。

    python
    from tqdm import tqdm
from time import sleep

pbar = tqdm(range(10))
for char in pbar:
    pbar.set_description('Processing %d' % char)
    sleep(0.5)
pbar.close()

# Processing 9: 100%|███████████████████| 10/10 [00:05<00:00,  1.71it/s]

  • 嵌套进度条

    python
    from tqdm import tqdm
import time

for top_level in tqdm(range(5), desc="first counter", total=5):
    for second_level in tqdm(range(2), desc="second counter", total=2):
        time.sleep(0.1)

# second counter: 100%|██████████████████| 2/2 [00:00<00:00,  9.62it/s]
# second counter: 100%|██████████████████| 2/2 [00:00<00:00,  9.50it/s]
# second counter: 100%|██████████████████| 2/2 [00:00<00:00,  9.54it/s]
# second counter: 100%|██████████████████| 2/2 [00:00<00:00,  9.64it/s]
# second counter: 100%|██████████████████| 2/2 [00:00<00:00,  9.72it/s]
# first counter: 100%|███████████████████| 5/5 [00:01<00:00,  4.79it/s]

  • 控制进度条大小

    python
    from tqdm import tqdm

for i in tqdm(range(999999), ncols=50):
    pass

# 100%|█| 999999/999999 [00:00<00:00, 5157511.49it/s

    仅展示百分比，将 ncols的值设置为 4。

    python
    from tqdm import tqdm
import time

for i in tqdm(range(999999), ncols=4):
    pass

# 100%

  • 禁用配置 disable

    python
    from tqdm import tqdm
import time

debug = False  # 通过修改 debug 配置控制进度条的展示与否
for i in tqdm(range(99999), disable=not debug):
    time.sleep(0.00001)

  • 在 notebook 中应用

    python
    from tqdm.notebook import tqdm
import time

for i in tqdm(range(99999)):
    time.sleep(0.00001)

    在 notebook 上展示的进度条带颜色。

    python
    from tqdm.notebook import tqdm
import time

counter = 0
for i in tqdm(range(99999)):
    if i % 2 == 0:
        counter += 1
    if counter == 30_000:
        break  # 跳出循环，进度条也将终止，进度条颜色变成红色，可以很明显的看到程序出错了

#  60%|████████████▌        | 59998/99999 [00:00<00:00, 2253533.67it/s]

  • 导入 tqdm.auto 自动判断使用的环境

    python
    from tqdm.auto import tqdm

手动控制进度 ​

手动运行时 tqdm 对象不需要迭代器作为初始化参数，但需要指定最大长度 total 的值，使用 update 方法自定义进度条比例，例如将 100 进度任务分为 5 次进度展示。：

python
from tqdm import tqdm
import time

with tqdm(total=100) as pbar:
    for i in range(1, 5):
        time.sleep(1)
        pbar.update(10 * i)

# 10%|███                            | 10/100 [00:01<00:09,  9.95it/s]True
# 30%|█████████▎                     | 30/100 [00:02<00:04, 15.81it/s]True
# 60%|██████████████████▌            | 60/100 [00:03<00:01, 22.24it/s]True
#100%|██████████████████████████████| 100/100 [00:04<00:00, 29.19it/s]True
#100%|██████████████████████████████| 100/100 [00:04<00:00, 24.90it/s]

通过 with 语法管理上下文，在循环体之外由编译器自动结束其生命周期。