Python中如何通过条件判断将print输出定向至文件或屏幕？

当然可以，有几种简洁的方法能避免重复写print语句，这里给你几个实用方案：

方案一：先确定输出流，统一执行print #

先根据条件判断输出目标是文件还是标准输出（屏幕），然后只调用一次print：

import sys
filename = 'log.txt'

# 确定输出对象：条件满足时打开文件，否则用sys.stdout（屏幕）
output = open(filename, 'a') if SomeCondition else sys.stdout
try:
    print("Hello World!", file=output)
finally:
    # 只有文件对象需要手动关闭，stdout不用处理
    if output is not sys.stdout:
        output.close()

这个方法逻辑清晰，能确保文件被正确关闭，避免资源泄漏。

方案二：用ExitStack统一管理上下文（更优雅） #

如果想利用with语句的自动资源管理特性，同时兼容两种输出场景，可以用contextlib.ExitStack：

from contextlib import ExitStack
import sys

filename = 'log.txt'

with ExitStack() as stack:
    if SomeCondition:
        # 把文件对象加入上下文栈，自动管理关闭
        output = stack.enter_context(open(filename, 'a'))
    else:
        output = sys.stdout
    # 统一执行一次print
    print("Hello World!", file=output)

这种写法不需要手动判断关闭文件，ExitStack会帮我们处理文件的自动关闭，代码更简洁安全。

方案三：动态设置print的file参数（极简写法） #

如果追求代码最短，可以直接在print的file参数里做条件判断，但要注意用with语句确保文件关闭：

filename = 'log.txt'

# 条件满足时用文件上下文，否则用None（对应stdout）
with open(filename, 'a') if SomeCondition else None as output:
    print("Hello World!", file=output)

这里None作为file参数时，print默认输出到屏幕，和直接调用print("Hello World!")效果一致，同时with语句会自动关闭文件（如果是文件的话）。

内容的提问来源于stack exchange，提问作者SG229