Python 如何将 stdout 重定向到文件

在本文中，我们将介绍如何使用 Python 将标准输出 (stdout) 重定向到文件。在开发过程中，通常我们在终端上打印输出信息以进行调试、观察和测试，但有时候我们希望将这些输出信息保存到文件中，以便后续查看和分析。Python 提供了几种方式来实现这一目标。

阅读更多：Python 教程

方法一：使用文件重定向符

在终端或命令行界面下，我们可以使用文件重定向符 > 将 stdout 输出重定向到一个文件中。具体的方法是，在运行 Python 脚本时，将输出命令以 > 连接一个文件名，即可将输出写入指定的文件。示例代码如下：

print("Hello, stdout redirect!")  # 输出到终端

执行以下命令，将 stdout 重定向到 output.txt 文件中：

python my_script.py > output.txt

执行完毕后，打开 output.txt 文件，即可看到 stdout 输出的内容。

这种方法简单易用，适用于将全部 stdout 输出写入文件的情况，但无法同时保存 stdout 和 stderr 的输出。如果希望保存 stderr 输出，需要使用其他方法。

方法二：使用 sys 模块重定向 stdout

Python 的 sys 模块提供了一个 stdout 对象，它代表了标准输出。我们可以通过修改这个对象来实现对 stdout 的重定向。示例代码如下：

import sys

# 保存原始 stdout 对象
original_stdout = sys.stdout

# 打开文件并将输出重定向到文件
with open('output.txt', 'w') as f:
    sys.stdout = f  # 将 sys.stdout 指向文件对象
    print("Hello, stdout redirect!")  # 输出到文件

# 恢复原始 stdout 对象
sys.stdout = original_stdout

# 输出到终端
print("Hello, back to terminal!")

在这个示例中，我们首先保存了原始的 stdout 对象，然后打开文件并将 sys.stdout 指向文件对象，这样所有的输出都将被重定向到文件中。最后，我们通过恢复原始的 stdout 对象来将输出恢复到终端。

这种方法更加灵活，可以将 stdout 输出重定向到文件，同时保留 stderr 的输出。但需要注意，对于使用 print 函数以外的方法输出的内容，这种方法可能无法完全适用。

方法三：使用 contextlib 模块重定向 stdout

Python 的 contextlib 模块提供了一个 redirect_stdout 上下文管理器，可以更方便地将 stdout 重定向到文件中。示例代码如下：

import contextlib

with open('output.txt', 'w') as f, contextlib.redirect_stdout(f):
    print("Hello, stdout redirect!")  # 输出到文件

print("Hello, back to terminal!")  # 输出到终端

在这个示例中，我们使用 with open 语句打开文件，并将 redirect_stdout 上下文管理器与文件对象连接。在 with 块内，所有的输出都将被重定向到文件中，而在 with 块外，输出将恢复到终端。

这种方法非常简洁，适用于将 stdout 输出重定向到文件的场景。同时也可以使用 contextlib.redirect_stderr(f) 将 stderr 输出重定向到同一文件中。

方法四：使用 subprocess 模块重定向 stdout

除了上述方法外，我们还可以使用 subprocess 模块来实现将 stdout 输出重定向到文件。示例代码如下：

import subprocess

# 执行命令并将 stdout 重定向到文件
subprocess.run(["python", "my_script.py"], stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True)

print("Hello, back to terminal!")  # 输出到终端

在这个示例中，我们使用了 subprocess.run 方法来执行命令，并通过设置 stdout 参数来将 stdout 输出重定向到 subprocess.PIPE 中。这样输出的结果将通过 subprocess.PIPE 返回给我们，我们可以将其保存到文件或其他地方进行处理。需要注意的是，使用 subprocess 模块需要先安装相应的依赖库。

这种方法适用于需要在 Python 脚本中调用其他脚本或命令并将输出保存到文件的情况。