如何在 Jupyter 中使用 tqdm 进度条？

Question

我正在使用 Jupyter 小部件编写一个界面，用户在其中配置一堆设置，然后单击“运行”按钮。单击后，此按钮将调用另一个模块中实现的函数。该函数需要一些时间才能运行，因此我使用该包向其添加了一个进度条tqdm。不幸的是，当我单击“运行”时，此进度条现在显示在另一个终端中，而不是在单元格的输出中。有没有办法让条形图显示在同一单元格输出中？

Answer 1

有一个tqdm与 jupyter 集成的子模块：https://github.com/tqdm/tqdm#ipython-jupyter-integration。如果您已经启用了小部件，它应该可以开箱即用。只需将任何 tqdm 调用替换为子模块中的等效调用tdqm.notebook，即 import

from tqdm.notebook import trange, tqdm

代替

from tqdm import trange, tqdm

这假设您实际上是直接在笔记本中调用 tqdm 并且该调用没有隐藏在其他模块代码中。如果是这种情况，您可能需要传递参数或修补 tqdm 调用。