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 hook function中。