📜  jupyter notebook 的 tqdm - Python (1)

📅  最后修改于: 2023-12-03 15:17:06.922000             🧑  作者: Mango

Jupyter Notebook使用tqdm优化Python代码的进度条显示

在进行数据处理、模型训练等任务时,我们通常需要显示一个进度条来显示任务的完成度。而Python中的tqdm库就提供了一种简单易用的方法来实现进度条的显示。

1. 安装tqdm

在使用前需要安装tqdm库,可以使用pip进行安装:

!pip install tqdm
2. 使用tqdm

使用tqdm非常简单,只需要将需要迭代处理的对象作为参数传入,并使用for循环进行迭代就可以了。

from tqdm import tqdm

for i in tqdm(range(100)):
    # 进行一些操作

这样就可以在Jupyter Notebook中看到一条进度条了。其中,range(100)是一个需要迭代处理的对象,tqdm默认将其划分为100份,每处理完一份就会刷新一次进度条。

除了range之外,其他的Python迭代对象也可以使用tqdm进行处理,比如列表、元组等。

from tqdm import tqdm

my_list = [1, 2, 3, 4, 5]

for item in tqdm(my_list):
    # 进行一些操作
3. 定制化进度条

tqdm库还提供了一些参数可以根据需要进行定制化。比如,可以设置进度条的颜色、长度、字符样式等。

from tqdm import tqdm

for i in tqdm(range(100), bar_format="{l_bar}{bar:20}{r_bar}{bar:-20b}"):
    # 进行一些操作

通过bar_format参数,我们可以定制化进度条的显示样式。"{l_bar}"表示进度条前面的内容,"{bar}"表示进度条本身,"20"表示进度条长度为20(可以设置成其他数字),"{r_bar}"表示进度条后面的内容,"{bar:-20b}"表示进度条使用“-”字符进行填充。

这里只是列举了一些比较简单的定制化参数,tqdm库还提供了更丰富的参数可以根据需要进行设置。

4. 总结

使用tqdm可以方便快捷地实现Python进度条的显示,不仅可以提高代码可读性,还可以让我们更好地监控代码的执行进度,提高开发效率。