python tqdm包显示进度条

Tqdm 是一个智能进度表，是一个快速，可扩展的 Python 进度条，可以在 Python长循环中添加一个进度提示信息，用户只需要封装任意的迭代器 tqdm(iterator)。它能够显示所有可迭代对象当前执行的进度。

1. 自动迭代

# pip install tqdm

from tqdm import tqdm
from time import sleep

# 查看参数
help(tqdm) 

for char in tqdm(["a", "b", "c", "d"]):
#for i in tqdm(range(10)):
    sleep(0.25)
    #do something
    
# 控制进度条显示当前步骤的名称
pbar = tqdm(["a", "b", "c", "d"])
for char in pbar:
    sleep(0.25)
    pbar.set_description("Processing %s" % char)

2. 手动迭代

import time
from tqdm import tqdm

with tqdm(total=100) as pbar:
    for i in range(10):
        sleep(0.1)
        pbar.update(10)

# 发呆0.5s
def action():
    time.sleep(0.5)
with tqdm(total=100000, desc='Example', leave=True, ncols=100, unit='B', unit_scale=True) as pbar:
    for i in range(10):
        # 发呆0.5秒
        action()
        # 更新发呆进度
        pbar.update(10000)

3. 显示下载文件进度

# -*- coding: utf-8 -*-
import os
import requests
from tqdm import tqdm
import os.path as op

# 文件url地址
file_url = "https://stringdb-static.org/download/protein.links.v11.5/9606.protein.links.v11.5.txt.gz"
# 提取文件名
filename = op.splitext(file_url)[-1]
# 保存文件位置及名称
file_to_save = op.join(os.getcwd(), filename)

response = requests.get(file_url, stream=True)
# 文件大小，字节/1024/1024=MB
print(response.headers)  # 打印查看基本头信息

data_size = round(int(response.headers['Content-Length'])) / 1024 / 1024

with open(file_to_save, "wb") as f:
    for data in tqdm(response.iter_content(chunk_size=1024*1024),total=data_size, desc='正在下载', unit='MB'):
        f.write(data)