如何根据Python中的if语句条件将打印命令输出重定向到文件或屏幕?

Question

我有一个打印字符串的简单代码。我想实现一个if条件，如果值为真，它将把字符串打印到文件中，否则它将打印到屏幕上。

有没有办法做到这一点，而没有2个相同的打印命令(一个打印到文件，另一个显示)?类似于下面的代码片段:

filename = 'log.txt'
if SomeCondition == TRUE:
with open(filename, 'a') as f:
print("Hello World!", file=f)
else:
print("Hello World!") #Is there a way to avoid duplicating the print statment?

谢谢你的帮助!

Answer 1

使用

filename = 'log.txt'
with open(filename, 'a') as f:
print("Hello World!", file=[None, f][condition])