IPython:将Python脚本的输出重定向到文件(如bash>)

Question

我有一个我想在IPython中运行的Python脚本.我想将输出重定向(写入)到文件,类似于:

python my_script.py > my_output.txt

当我在IPython中运行脚本时,我该怎么做呢,比如说 execfile('my_script.py')

有一个较旧的页面描述了可以编写的函数来执行此操作,但我相信现在有一种内置的方法来执行此操作,我找不到.

Answer 1

IPython有自己的上下文管理器来捕获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示例笔记本中所示.

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

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

Answer 2

编写一个脚本对我来说似乎有些过分,因为我只是想在IPython中工作时看到变量中包含的大量文本.这对我有用:

%store VARIABLE >> file.txt (附加)
%store VARIABLE > file.txt(覆盖)