欢迎访问宙启技术站
智能推送

快速入门:学习使用tqdm()库来实现进度条显示

发布时间:2023-12-28 12:14:40

快速入门:使用tqdm()库实现进度条显示

在编写长时间运行的代码时,为了让用户了解代码的执行进展,我们通常会希望在终端中显示一个进度条。tqdm()是一个快速、可扩展的Python进度条库,可以帮助我们实现这个目标。

安装和导入

首先,我们需要使用pip命令安装tqdm库。打开终端,执行以下命令:

pip install tqdm

安装完成后,我们可以在代码中导入tqdm模块:

from tqdm import tqdm

使用tqdm()函数创建一个进度条

使用tqdm()非常简单。只需对需要迭代的对象调用tqdm(),并传递给它即可:

my_list = range(100000)
for item in tqdm(my_list):
    # 执行你的代码

在这个例子中,假设我们有一个包含100,000个元素的列表。通过调用tqdm(my_list),我们将创建一个与列表长度相对应的进度条。在每次迭代时,进度条会显示已完成的百分比。

设置进度条描述

你还可以设置一个文本描述,用于描述进度条正在进行的任务。只需将描述字符串作为tqdm()的第二个参数传递进去:

my_list = range(100000)
for item in tqdm(my_list, desc="Processing"):
    # 执行你的代码

在这个例子中,进度条会显示为"Processing",以告诉用户代码正在处理的是什么。

设置进度条样式

tqdm()还提供了一些可供选择的样式选项,以改变进度条的外观。你可以选择不同的样式,以适应不同的编程环境。以下是一些可选的样式参数:

+ "tqdm":默认样式,显示进度百分比和时间估计。

+ "bar":只显示进度条,不显示任何其他统计信息。

+ "simple":更简洁的样式,只显示进度百分比。

+ "custom_style":自定义进度条样式。

要使用不同的样式,只需将style参数传递给tqdm()函数:

my_list = range(100000)
for item in tqdm(my_list, desc="Processing", bar_format="{desc}: {percentage:.2f}%|{bar}"):
    # 执行你的代码

在这个例子中,"bar_format"参数定义了进度条的样式。我们使用了"{desc}: {percentage:.2f}%|{bar}"格式,它会在进度条前面显示描述和百分比。你可以根据自己的需要自定义样式。

更多选项

tqdm()还提供了其他一些选项,以进一步定制进度条的行为。例如,使用total参数可以指定迭代的总数,这将为进度条提供一个确切的完成时间估计。tqdm还支持在进度条上显示速度、剩余时间和迭代次数等信息。要了解更多选项,请参阅tqdm文档。

总结

使用tqdm()库可以很容易地实现进度条显示。通过简单的调用,我们可以创建具有不同样式和描述的进度条,并将其应用于我们的代码中的迭代过程。这提供了一种简单而强大的方式,让我们的代码在终端中显示进度,以便用户可以了解代码的执行进展。