要捕获函数的标准输出并将其写入文件,可以使用重定向操作符>
将标准输出重定向到一个文件中。以下是一个示例代码:
import sys
def my_function():
print("Hello, World!") # 要捕获的输出
# 将标准输出重定向到文件
with open('output.txt', 'w') as f:
sys.stdout = f # 将标准输出重定向到文件对象
# 调用函数
my_function()
# 恢复标准输出
sys.stdout = sys.__stdout__
# 输出已被写入文件
print("Output has been captured and written to file.")
在上述示例中,首先将标准输出重定向到一个文件对象f
,然后调用函数my_function()
,其输出将被写入文件。最后,恢复标准输出,以便后续的输出可以显示在控制台上。
请注意,重定向标准输出后,所有的print
语句都会将输出写入到文件中,而不是显示在控制台上。如果你希望只捕获特定函数的输出,可以将重定向和恢复标准输出的代码放置在需要捕获输出的函数之前和之后。
下一篇:捕获函数的结果