4

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

tqdm在使用 usingJupyter Notebook和 encoutering a运行循环时RunTime Error，恢复确实很难 - 修复错误并重新运行循环会导致多行打印，而不是tqdm原始的单行打印。恢复的唯一方法是重新启动内核，这不是很有用。

还有其他解决方案吗？

我无法故意重现该问题，但这里有一个可能导致该问题的示例代码：

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

然后重新运行单元格。

4

1 回答 1

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]
于 2017-09-29T11:28:49.877 回答