tqdm：Python 中的进度条工具

tqdm 是 Python 中一个简单易用的进度条工具，可以方便地监控代码的运行进度。它可以在循环体中使用，为我们的数据处理提供视觉上的辅助。我们不再需要等待所有的数据处理完成之后才能获得结果。

安装

使用 pip 可以很方便地安装 tqdm。

pip install tqdm

示例

在 Python 的 for 循环中使用 tqdm。

from tqdm import tqdm
import time

for i in tqdm(range(10)):
    time.sleep(0.5)

在 Pandas 的 DataFrame 中使用 tqdm。

import pandas as pd
from tqdm import tqdm

df = pd.read_csv('data.csv')

for index, row in tqdm(df.iterrows(), total=df.shape[0]):
    # do something

进度条样式

通过修改 tqdm 的参数，可以修改进度条的样式。

设置进度条样式为绿色：

tqdm(range(100), bar_format='{l_bar}{bar:30}{r_bar}', colour='green')

设置进度条样式为蓝色：

tqdm(range(100), bar_format='{l_bar}{bar:30}{r_bar}', colour='blue')

注意事项

不要在 Jupyter Notebook 中使用 tqdm 自动关闭进度条。

tqdm_notebook(...)

如果需要在 Jupyter Notebook 中使用 tqdm，请改用以下方式：

from tqdm.auto import tqdm
tqdm.pandas()
df.progress_apply(func)

这样可以避免自动关闭进度条的问题。