將任意函數的輸出轉換爲列表

Question

我有一個函數。它打印字符串。我怎樣才能最好地收集列表中的這些字符串？

即

def FUN(a): 
... 
    print(somestrings) 

Answer 1

x = list() 

x.append(strings1) # use a loop of some sort to append your strings 
x.append(strings2) 
. 
. 
. 
x.append(stringsN) 

return x 

Answer 2

你可以臨時更改sys.stdout到自定義類：

import sys 

def fun(a,b): 
    print(a) 
    print(b) 

class Catcher: 
    def __init__(self): 
     self.res = [] 
     self.stdout = sys.stdout 
     sys.stdout = self 
    def write(self, chars): 
     self.res.append(chars) 
    def call(self, fun, *args, **kwargs): 
     fun(*args, **kwargs) 
     sys.stdout = self.stdout 
     return self.res 

print(Catcher().call(fun, "hello", b="world!")) 

=> ['hello', '\n', 'world!', '\n']