您的位置:

使用Python tqdm让你的代码进度条更丰富有趣

在编写长时间运行的代码时,我们常常会使用进度条来实时显示任务的进度。Python提供了许多实现进度条的库,其中tqdm是一个高度可定制的库,它可以让你的代码进度条更加丰富有趣。

一、简介

tqdm是一个Python的进度条库,它可以在循环迭代时实时显示任务的进度,支持多种方式的进度条样式,可以自定义进度条的位置、颜色、速度等。安装tqdm库十分简单,只需在终端输入以下命令:

    pip install tqdm

安装完成后,我们可以直接在代码中使用它,来创建漂亮的进度条。

二、基础使用

让我们首先看一下最简单的使用方式。下面的代码演示了tqdm在循环中的基本用法:

    # 导入tqdm模块
    from tqdm import tqdm
    
    # 执行循环
    for i in tqdm(range(1000)):
        # 这里是循环的操作

以上代码使用了tqdm工具实现了一个简单的进度条。通过tqdm函数,我们将range(1000)传递进去,然后在for循环中使用tqdm遍历range(1000)中的元素。tqdm将自动显示进度条,并实时更新进度。我们还可以仅仅使用tqdm函数,如下所示:

    # 导入tqdm模块
    from tqdm import tqdm
    
    # 执行循环
    for i in tqdm(range(1000)):
        # 这里是循环的操作

此时会使用默认样式显示进度条,如下所示:

  5%|███▏                                                                             | 50/1000 [00:00<00:01, 489.96it/s]

默认样式比较简单,只有一个进度条,但我们可以使用一些其他的样式来更好地展示进度条,后面会提到。

三、自定义样式

tqdm是一个高度可定制的库,你可以使用各种可用参数来定制进度条。下面介绍一些可用的参数:

  • total:表示迭代次数。
  • position:进度条的位置。
  • desc:进度条的描述信息。
  • unit:进度条每个单位的描述信息。
  • bar_format:进度条的样式,可以自定义。

下面的代码演示了如何使用以上参数进行自定义。

    # 导入tqdm模块
    from tqdm import tqdm
    
    # 使用自定义样式
    with tqdm(total=100, position=0, desc='Downloading', unit='file') as pb:
        for i in range(100):
            pb.update(1)

以上代码定义了进度条总共需要迭代100次,进度条显示在第一行,进度条的描述为'Downloading',每个单位为'file'。而pb.update(1)则表示迭代次数加1。这段代码的效果如下:

Downloading: 100%|█████████████████████████████████████████████████████████████████████| 100/100 [00:00<00:00, 4811.35it/s]

可以看到,我们自定义的样式不仅包含了进度条,还有进度条当前的百分比,已处理的文件数、总文件数、处理速度等信息。所有这些信息都可以在bar_format这个参数中进行自定义。

四、更多自定义样式

tqdm提供了许多内置的进度条样式,我们可以从中选择适合自己项目的样式。下面是一些常见的样式:

  • 'bar': 进度条样式,如[=== ]
  • 'barh': 水平样式,如[= ]
  • 'progress': 进度和百分比
  • 'percentage': 进度百分比
  • 'count': 进度条计数
  • 'n': 显示n的当前值

在我们的代码中,可以设置如下参数进行自定义样式:

    # 导入tqdm模块
    from tqdm import tqdm
    
    # 使用进度计数样式
    with tqdm(range(1000), desc='Processing', unit='document', bar_format='{l_bar}{bar:20}{r_bar}{bar:-20b}') as pb:
        for i in pb:
            # 这里是循环的操作
        pb.update(1)

以上代码使用了默认的样式,但是在此基础上设置了进度条的左边缘、进度条、右边缘和进度条两端之间的长度分别为1、20、1、20。效果如下:

Processing: 100%|████████████████████████████████████████████████████████████████| 1000/1000 [00:00<00:00, 44730.08document/s]

除了基本样式之外,我们还可以使用tqdm提供的更丰富的样式,如下所示:

    # 导入tqdm模块
    from tqdm import tqdm
    
    # 使用设置颜色样式
    with tqdm(range(100), bar_format='\x1b[32m{bar}\x1b[0m') as pb:
        for i in pb:
            # 这里是循环的操作
            pb.set_postfix({'process': i})

以上代码使用了绿色的进度条样式,并且在进度条右边显示了当前进度。效果如下:

 35%|██████████████████████████▋                                                | 35/100 [00:00<00:00, 343.92it/s, process=35]

可以看到,我们的进度条不仅可以改变颜色,还可以显示进度条右边的其他信息。

五、总结

使用Python tqdm库可以让你的代码进度条更加丰富多彩,许多内置的样式可以让我们自定义不同的进度条样式。通过tqdm,我们能够更好地把握任务进度,让编写代码更加愉快。

如果想要学习更多关于tqdm的使用,可以参考它的官方文档: https://github.com/tqdm/tqdm