python显示代码执行进度

原创

mob64ca12f3496a 2024-01-19 09:32:28 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12f3496a的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python显示代码执行进度

在Python中，我们经常会遇到需要执行耗时任务的情况。这些任务可能是数据处理、模型训练、网络请求等等。当任务执行时间较长时，我们希望能够实时地查看代码的执行进度，以便于了解任务的进展情况。本文将介绍一种常见的方法，即使用tqdm库来显示代码的执行进度。

什么是tqdm

tqdm是一个Python库，用于在命令行界面中显示进度条。它可以轻松地在循环中添加一个进度条，并自动估计剩余时间。使用tqdm可以更直观地了解代码的执行进度，提高开发效率。

安装tqdm

要使用tqdm库，首先需要安装它。可以通过以下命令使用pip安装：

pip install tqdm

使用tqdm显示进度条

使用tqdm库显示进度条非常简单。我们只需要在循环中使用tqdm函数来包装迭代对象即可。下面是一个简单的示例：

from tqdm import tqdm
import time

# 创建一个迭代对象，例如一个列表或者一个range对象
data = range(100)

# 使用tqdm包装迭代对象，并设置描述信息
for i in tqdm(data, desc='Processing'):
    # 模拟耗时操作
    time.sleep(0.1)

在上面的代码中，我们首先导入了tqdm库和time库。然后，我们创建了一个包含100个元素的range对象，并使用tqdm函数将其包装起来。在循环中，我们使用time.sleep函数模拟了一个耗时操作。在每次循环迭代时，tqdm会自动更新进度条并显示剩余时间。

运行上面的代码，我们可以看到在命令行界面中显示了一个进度条，同时也显示了剩余时间。这样我们就能够实时地了解代码的执行进度了。

进度条的其他设置项

除了基本的使用方法，tqdm还提供了一些其他的设置项，以便于我们根据实际需求进行配置。

ncols：指定进度条的宽度。
unit：指定进度条的单位，默认是“it/s”（每秒迭代次数）。
bar_format：指定进度条的格式。可以使用{desc}、{percentage}、{r_bar}等占位符来自定义格式。
leave：设置为True时，进度条完成后会保留在屏幕上。

下面是一个示例，演示了如何使用这些设置项：

from tqdm import tqdm
import time

data = range(100)

for i in tqdm(data, desc='Processing', ncols=80, unit='item', bar_format='{l_bar}{bar}| {percentage:3.0f}%'):
    time.sleep(0.1)

在上面的代码中，我们通过设置ncols参数指定了进度条的宽度为80个字符，设置unit参数为“item”，并使用了自定义的进度条格式。