溫馨提示×

如何在Python中處理print的輸出流

小樊
84
2024-08-29 06:45:50
欄目: 編程語言

在Python中,您可以通過更改sys.stdout來處理print函數(shù)的輸出流

import sys

# 保存原始stdout引用,以便稍后恢復(fù)
original_stdout = sys.stdout

with open("output.txt", "w") as output_file:
    # 將stdout重定向到文件
    sys.stdout = output_file
    
    print("這個(gè)輸出將會(huì)被寫入output.txt文件。")

# 恢復(fù)原始stdout
sys.stdout = original_stdout

print("這個(gè)輸出將會(huì)正常顯示在控制臺(tái)上。")

在這個(gè)例子中,我們首先導(dǎo)入了sys模塊,然后將原始的stdout引用保存到original_stdout變量中。接下來,我們使用with open()語句打開一個(gè)名為output.txt的文件,并將其設(shè)置為寫入模式("w")。在with語句塊內(nèi)部,我們將sys.stdout重定向到output_file,這樣所有的print語句都會(huì)將輸出寫入output.txt文件。在with語句塊之外,我們將sys.stdout恢復(fù)為原始值,這樣print函數(shù)就可以正常地在控制臺(tái)上顯示輸出了。

0