使用tqdm库在Python中优雅地显示进度条

01_6

于 2024-07-28 16:26:32 发布

阅读量64

点赞数 4

文章标签： python 开发语言人工智能深度学习 pytorch 机器学习

本文链接：https://blog.csdn.net/2301_77444219/article/details/140752191

版权

在Python编程中，当我们处理长时间运行的循环或任务时，了解任务的进度对于用户来说是非常有帮助的。tqdm是一个流行的Python库，它可以帮助我们在命令行中方便地显示进度条。本文将介绍如何使用tqdm库来增强你的Python脚本和项目的用户体验。

首先，你需要在你的Python环境中安装tqdm库。这可以通过pip包管理器轻松完成。打开你的命令行工具（WIN+R，输入CMD，回车），然后输入以下命令：

pip install tqdm

这个命令会从Python包索引（PyPI）下载并安装tqdm库。

from tqdm import tqdm

tqdm库还提供了一个方便的trange函数，它是range和tqdm的结合体，用于直接在for循环中创建进度条。

你可以将tqdm()函数包装在任何可迭代的对象上，比如range()或列表（list）。这样做会在迭代时显示进度条。以下是一个简单的示例：

import time  
from tqdm import tqdm  
  
# 使用tqdm包装range  
for i in tqdm(range(100)):  
    # 模拟耗时操作  
    time.sleep(0.1)

tqdm(range(100))创建了一个进度条对象，它会迭代100次。每次迭代都会更新进度条，并且我们通过time.sleep(0.1)来模拟耗时操作。

trange()是tqdm库提供的一个便捷函数，它结合了range()和tqdm()的功能。你可以像使用range()一样使用它，但会自动显示进度条：

import time  
from tqdm import trange  
  
# 使用trange代替range  
for i in trange(100):  
    # 模拟耗时操作  
    time.sleep(0.1)

这个示例和上一个示例的效果相同，但代码更简洁。

关注