IPython：将Python脚本的输出重定向到文件（比如bash>）

Question

我有一个Python脚本要在IPython中运行。我想将输出重定向（写入）到一个文件，类似于：

python my_script.py > my_output.txt

如何在IPython中运行脚本，例如execfile('my_script.py')

有一个older page描述了一个函数，它可以被编写来完成这个任务，但是我相信现在有一个内置的方法来完成这个任务，我只是找不到。

Answer 1

IPython有自己的capturing stdout/err上下文管理器，但它不重定向到文件，而是重定向到一个对象：

from IPython.utils import io
with io.capture_output() as captured:
    %run my_script.py

print captured.stdout # prints stdout from your script

这个功能在%%capture单元魔术中公开，如Cell Magics example notebook所示。

这是一个简单的上下文管理器，因此您可以编写自己的版本来重定向到文件：

class redirect_output(object):
    """context manager for reditrecting stdout/err to files"""


    def __init__(self, stdout='', stderr=''):
        self.stdout = stdout
        self.stderr = stderr

    def __enter__(self):
        self.sys_stdout = sys.stdout
        self.sys_stderr = sys.stderr

        if self.stdout:
            sys.stdout = open(self.stdout, 'w')
        if self.stderr:
            if self.stderr == self.stdout:
                sys.stderr = sys.stdout
            else:
                sys.stderr = open(self.stderr, 'w')

    def __exit__(self, exc_type, exc_value, traceback):
        sys.stdout = self.sys_stdout
        sys.stderr = self.sys_stderr

你可以用它来调用：

with redirect_output("my_output.txt"):
    %run my_script.py