Python三种标准输出重定向方法

Python 提供了标准库中的 sys.stdout 对象来代表标准输出。如果我们想将输出重定向到其他位置，例如内存、文件或自定义类，我们可以通过一些技巧来实现。

1. 重定向到内存

要将输出重定向到内存，我们可以使用 StringIO 模块。下面是一个示例： 

import sys
from io import StringIO

# 创建一个字符串缓冲区来替代默认的标准输出
buffer = StringIO()
sys.stdout = buffer

# 在这里执行需要输出的代码
print("Hello, World!")
print("This is a test.")

# 获取重定向后的内容
output = buffer.getvalue()

# 还原标准输出
sys.stdout = sys.__stdout__

# 打印重定向后的结果
print(output)

 

在上面的示例中，我们首先创建了一个 StringIO 对象 buffer，然后将 sys.stdout 重定向到该对象。接着，我们执行一些输出操作，并使用 getvalue() 方法获取重定向后的内容。最后，我们还原标准输出，并打印出重定向后的结果。

2. 重定向到文件

如果我们想将输出重定向到文件，可以使用 sys.stdout 对象和文件操作符。下面是一个示例：

import sys

# 打开一个文件以写入输出
with open("output.txt", "w") as f:
    # 保存原来的标准输出
    original_stdout = sys.stdout
    # 重定向标准输出到文件
    sys.stdout = f

    # 在这里执行需要输出的代码
    print("Hello, World!")
    print("This is a test.")

    # 还原标准输出
    sys.stdout = original_stdout

 

在上面的示例中，我们首先打开一个文件，在 "w" 模式下用于写入输出内容。接着，我们保存了原来的标准输出对象 sys.stdout，并将其重定向到文件 f。然后，我们执行一些输出操作，并还原标准输出。

3. 重定向到自定义类

如果我们想将输出重定向到自定义的类中，可以创建一个类，并覆盖该类的 write() 方法，使其接收要输出的内容。下面是一个示例：

import sys
import io

class CustomWriter(io.TextIOBase):
    def __init__(self):
        self.content = ""

    def write(self, text):
        self.content += text

# 创建一个自定义的输出写入器
custom_writer = CustomWriter()

# 保存原来的标准输出
original_stdout = sys.stdout
# 重定向标准输出到自定义的写入器
sys.stdout = custom_writer

# 在这里执行需要输出的代码
print("Hello, World!")
print("This is a test.")

# 还原标准输出
sys.stdout = original_stdout

# 打印重定向后的结果
print(custom_writer.content)

 在上面的示例中，我们首先创建了一个自定义类 CustomWriter，需要继承io.TextIOBase。该类具有一个 write() 方法用于接收输出内容。接着，我们创建一个自定义的输出写入器 custom_writer，并将其赋值给 sys.stdout，以将输出重定向到该写入器。然后，我们执行一些输出操作，并最后还原标准输出。