如何使用Python将print打印的内容存储到txt文件|极客教程

如何使用Python将print打印的内容存储到txt文件

在Python编程中，经常会遇到需要将print打印的内容存储到txt文件中的需求。这种操作在调试程序、记录日志或输出等情况下非常有用。在本文中，我将详细介绍如何使用Python实现将print打印的内容存储到txt文件的方法。

1. 使用重定向

Python的sys模块提供了一种简单的方法来重定向标准输出流，从而实现将print输出的内容存储到txt文件中。下面是具体的步骤：

import sys

# 定义输出重定向函数
def redirect_stdout(file_name):
    sys.stdout = open(file_name, 'w')

# 将打印内容存储到txt文件中
redirect_stdout('output.txt')
print('Hello, world!')
print('This is a test.')

# 恢复标准输出流
sys.stdout = sys.__stdout__

在上面的代码中，首先定义了一个redirect_stdout函数，该函数接受一个文件名作为参数，并将标准输出流重定向到该文件。然后调用redirect_stdout函数，将print输出的内容存储到output.txt文件中。最后通过将标准输出流恢复到原来的状态，确保程序的正常输出。

2. 使用contextlib库

除了使用sys模块，还可以使用Python的contextlib库来实现将print打印的内容存储到txt文件中的功能。下面是一个使用contextlib.redirect_stdout上下文管理器的示例：

from contextlib import redirect_stdout

# 将打印内容存储到txt文件中
with open('output.txt', 'w') as f:
    with redirect_stdout(f):
        print('Hello, world!')
        print('This is a test.')