Jupyter Notebook中使用tqdm失败

Question

tqdm是我最喜欢的Python包之一，但是我遇到了一个烦人的问题，并试图弄清楚这是否是我的问题。

在使用Jupyter Notebook和tqdm运行循环时，遇到RunTime Error导致恢复非常困难 - 修复错误并重新运行循环后，通常会出现多行打印，而不是tqdm原始的单行打印。唯一的方法是重新启动Kernel，这不是非常有用。

有其他解决方案吗？

我无法故意复制此问题，但以下是可能导致该问题的示例代码：

from tqdm import trange
s=0
for i in trange(100):
    s+=i
    if i==10:
        raise ValueError

然后重新运行该单元格。

- Dimgold

---

你解决过这个问题吗？每当我中断一个单元格（实际上是KeyboardInterrupt），我都会遇到类似的问题。 - Josh Bode

Answer 1

为了解决这个问题，我使用以下的条件import，它适用于可能在Jupyter Notebook中运行的代码，或者在控制台中运行（如果不是交互式运行，则不执行任何操作）:

# progress.py
import sys


def tqdm(iterable, **kwargs):
    """Fake progress function."""

    return iterable


# check if running notebook and use notebook backend for tqdm progress bar
if 'IPython' in sys.modules:

    from IPython import get_ipython
    from IPython.display import display_javascript, Javascript

    from tqdm import tqdm

    ip = get_ipython()

    if 'IPKernelApp' in ip.config:
        monkey_patch = f"""
            from tqdm import tqdm_notebook as tqdm
            import {__name__} as module
            module.tqdm = tqdm
        """
        monkey_patch = ';'.join(x.strip() for x in monkey_patch.strip().split('\n'))
        display_javascript(Javascript(f"""IPython.notebook.kernel.execute("{monkey_patch}");"""))

然后使用它：

$ jupyter console  # or ipython3
In [1]: import progress

In [2]: list(progress.tqdm(range(10)))
100%|██████████| 10/10 [00:00<00:00, 120873.31it/s]
Out[2]: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

- Josh Bode