python tqdm_Python 进度条: tqdm

Tqdm 是 Python 进度条库，可以在 Python 长循环中添加一个进度提示信息。

pip install tqdm

1. 传入可迭代对象

import time
from tqdm import tqdm
for i in tqdm(range(100)):
    time.sleep(0.01)

trange(i) 是 tqdm(range(i)) 的简单写法

import time
from tqdm import trange
for i in trange(100):
    time.sleep(0.01)

输出:

100%|██████████| 100/100 [00:01<00:00, 97.68it/s]

2. 为进度条设置描述

在 for 循环外部初始化 tqdm，可以打印其他信息：

import time
from tqdm import tqdm

pbar = tqdm(["a", "b", "c", "d"])
for char in pbar:
    # 设置描述
    pbar.set_description("Processing %s" % char)
    time.sleep(1)

输出:

Processing d: 100%|██████████| 4/4 [00:04<00:00,  1.00s/it]

3. 手动控制进度：

import time
from tqdm import tqdm

# 一共200个，每次更新10，一共更新20次
with tqdm(total=200) as pbar:
    for i in range(20):
        pbar.update(10)
        time.sleep(0.1)

输出:

100%|██████████| 200/200 [00:02<00:00, 99.70it/s]

4. tqdm 的 write 方法

import time
from tqdm import tqdm, trange
for i in trange(10):
    time.sleep(0.1)
    if not (i % 3):
        tqdm.write("Done task %i" % i)

输出:

10%|█         | 1/10 [00:00<00:00,  9.96it/s]Done task 0
 40%|████      | 4/10 [00:00<00:00,  9.95it/s]Done task 3
 70%|███████   | 7/10 [00:00<00:00,  9.95it/s]Done task 6
100%|██████████| 10/10 [00:01<00:00,  9.94it/s]
Done task 9