python:从函数内部的“print”获取输出

Question

我使用一个函数库，其中一些函数打印我需要的数据：

def func():
   print "data"

如何调用此函数并将打印的数据转换为字符串？在

Answer 1

如果无法更改这些函数，则需要重定向sys.stdout：

>>> import sys
>>> stdout = sys.stdout
>>> import StringIO
>>> s = StringIO.StringIO()
>>> sys.stdout = s
>>> print "hello"
>>> sys.stdout = stdout
>>> s.seek(0)
>>> s.read()
'hello\n'