如何在Google Colab中将Python单元格输出重定向并打印到文件中？

单单元格定向捕获（无全局副作用） #

适合仅需要保存指定代码块的输出、不影响后续其他单元格运行逻辑的场景，借助Python标准库的重定向能力实现，无需安装第三方依赖：

• 导入需要的标准库组件

  from contextlib import redirect_stdout, redirect_stderr
  import io
  

• 两种捕获写法可选：
  1. 保留Colab单元格原有实时输出，同时把所有内容同步写入指定文件

     # 提前指定要保存的文件路径
     log_save_path = "cell_output.log"
     buffer = io.StringIO()
     # 把代码块的stdout、stderr全部捕获到内存缓冲区
     with redirect_stdout(buffer), redirect_stderr(buffer):
         # 此处替换为你要运行的所有业务代码：print语句、函数调用、报错栈、库打印信息都会被捕获
         print("测试输出内容")
     # 把捕获到的内容写入目标文件
     with open(log_save_path, "w", encoding="utf-8") as f:
         f.write(buffer.getvalue())
     

  2. 完全静默单元格输出，所有内容直接写入文件不在Colab页面显示

     log_save_path = "silent_output.log"
     with open(log_save_path, "w", encoding="utf-8") as log_file:
         with redirect_stdout(log_file), redirect_stderr(log_file):
             # 此处的代码输出不会显示在Colab单元格，直接写入目标文件
             print("静默测试输出")
     

优势说明：捕获范围严格限定在上下文管理器的代码块内，运行结束后自动恢复系统默认的输出流，不会污染全局Python环境；同时覆盖stdout和stderr通道，不会遗漏函数运行抛出的报错、警告类输出。

全局双写重定向（跨单元格自动存档） #

适合长流程任务：从某一步开始，后续所有单元格的运行输出都自动同步写入日志文件，不需要每个单元格重复写捕获逻辑。采用双写的Tee机制，不会丢失Colab页面的实时显示效果：

import sys
# 保存系统原生的输出流对象，方便后续随时恢复
original_stdout = sys.stdout
original_stderr = sys.stderr
# 打开目标日志文件，采用追加模式避免覆盖已有内容
log_file = open("global_run.log", "a", encoding="utf-8")

# 自定义双写流类，实现同时向多个输出通道写内容
class DualStream:
    def __init__(self, *target_streams):
        self.streams = target_streams
    def write(self, msg):
        for stream in self.streams:
            stream.write(msg)
    def flush(self):
        for stream in self.streams:
            stream.flush()

# 替换全局输出流，后续所有输出都会同时在Colab页面显示+写入日志文件
sys.stdout = DualStream(original_stdout, log_file)
sys.stderr = DualStream(original_stderr, log_file)

需要恢复正常输出时，执行以下代码即可切回原系统流：

sys.stdout = original_stdout
sys.stderr = original_stderr
log_file.close()

零代码补存已有单元格输出 #

如果不需要提前配置捕获逻辑，Colab本身内置了单元格输出导出能力：找到已运行完成的目标单元格，点击输出区右上角的三点菜单，选择下载输出选项，即可将该单元格的全部运行结果（文本、图表、报错信息）保存到本地。
适用场景：适合临时补存已经跑完的代码输出，不需要改动任何原有代码，但无法跨单元格合并多段输出到同一个文件。

实用注意事项 #

• 打开文件写入时务必指定encoding="utf-8"，避免中文、特殊字符写入日志时出现乱码
• 长流程任务建议用追加模式"a"打开日志文件，避免单元格重跑时覆盖之前的历史输出
• 若需要捕获第三方库的初始化输出（比如框架加载、环境检测信息），需要把重定向逻辑放在库导入代码之前执行，否则导入阶段的输出不会被捕获

内容的提问来源于stack exchange，提问作者Rafael Lourenço