Jupyter 笔记本输出重定向

Question

我想重定向笔记本每个单元格的输出。这是我尝试过的class Logger():    def __init__(self, stdout):        self.stdout = stdout    def write(self, msg):        self.stdout.write('augmented:' + msg)    def flush(self):        self.stdout.flush() 在一个单元格中，我stdout即时更改sys.stdout = Logger(sys.stdout)但是，下一个执行单元格的输出字符串没有“增强”字符串

Answer 1

您可以使用contextlib.

from contextlib import contextmanager

import sys

@contextmanager

def stdout_redirector():

    class MyStream:

        def write(self, msg):

            prefix = 'augmented:' if msg.strip() else ''

            old_stdout.write( prefix + msg)

        def flush(self):

            old_stdout.flush()

    old_stdout = sys.stdout

    sys.stdout = MyStream()

    try:

        yield

    finally:

        sys.stdout = old_stdout

最好使用with语句来管理重定向。如果在您的情况下这是不可能的，则调用重定向器对象的__enter__()和__exit__()方法也可以。您还可以将这些重定向器代码放在 IPython的pre_run_cell和post_run_cell 钩子函数中。