我正在嘗試創(chuàng)建一種機(jī)制將輸出重定向print到多個(gè)變量。以下代碼模擬了我正在尋找的內(nèi)容:-import sysimport ioclass MultiOut(object): def __init__(self, stream_out): self.terminal = sys.stdout self.output = stream_out def write(self, message): self.terminal.write(message) self.output.write(message) def flush(self): self.terminal.flush() self.output.flush()vals = {'a1': io.StringIO(), 'a2': io.StringIO(), 'a3': io.StringIO()}for i,val in enumerate(vals): sys.stdout = MultiOut(vals[val]) [print(x**i, end=' ') for x in range(11)] print("\n")with open('temp.txt', 'w') as f: for x in vals: f.write(f"{x} :-\n") f.write(vals[x].getvalue()) f.write(f"{'='*50}\n")文件輸出(tmp.txt):-a1 :-1 1 1 1 1 1 1 1 1 1 1 0 1 2 3 4 5 6 7 8 9 10 0 1 4 9 16 25 36 49 64 81 100 ==================================================a2 :-0 1 2 3 4 5 6 7 8 9 10 0 1 4 9 16 25 36 49 64 81 100 ==================================================a3 :-0 1 4 9 16 25 36 49 64 81 100 ==================================================我在這里想做的是將輸出的各個(gè)“部分”重定向到不同的變量vals - a1, a2, a3并將所有輸出轉(zhuǎn)儲(chǔ)到終端。奇怪的是,每個(gè)連續(xù)變量都包含從該點(diǎn)開始直到結(jié)束的數(shù)據(jù)。有沒有辦法避免這個(gè)問題并將每個(gè)部分保存在不同的變量中?
1 回答

慕田峪7331174
TA貢獻(xiàn)1828條經(jīng)驗(yàn) 獲得超13個(gè)贊
問題是您正在用您的對(duì)象替換 sys.stdout :
sys.stdout = MultiOut(vals[val])
在你的對(duì)象初始化中,你將 sys.stdout 設(shè)置為對(duì)象屬性
self.terminal = sys.stdout
在第二次迭代時(shí),執(zhí)行此語句時(shí)
self.terminal = sys.stdout
sys.stdout 是第一次迭代的替換,其中包括第一個(gè)MultiOut
對(duì)象。
我希望這是有道理的。
我會(huì)使用日志記錄模塊來實(shí)現(xiàn)您想要的,而不是修改標(biāo)準(zhǔn)輸出。
添加回答
舉報(bào)
0/150
提交
取消