Python捕获一个函数的输出并将其作为变量使用
原创
在 Python 中，可以通过多种方法捕获一个函数的输出并将其赋值给变量。具体方法取决于输出是函数返回的值，还是标准输出（print）输出的内容。以下是两种情况的解决方案：
1、问题背景
如果您有一个函数包含大量 print 语句，您希望该函数的执行结果存储在变量中，以便稍后使用，而不是直接输出到控制台。 例如，以下是一个函数：
def funA():
print("Hi")
print("There")
print("Friend")
print("!")我们希望能够像这样使用它：
def main():
# funA() 不会在控制台输出任何内容
a = getPrint(funA()) # getPrint 是一个假设的函数/对象
print(a) # 打印与执行 funA() 相同的结果这样当 funA 被调用时，它不会在控制台输出任何内容，而是将结果输出到一个对象中。我们随后再打印该对象以获取结果。
2、解决方案
我们可以使用 cStringIO 模块来实现这一目标。首先，在 getPrint 函数中，我们可以将系统标准输出（即 sys.stdout）重定向到一个 StringIO 对象。然后调用要捕获输出的函数，最后再将标准输出重定向回原来的位置。这样，就可以捕获函数的输出并将其作为字符串返回。例如，以下是如何使用 cStringIO 模块捕获函数输出的示例：
import cStringIO
import sys
def getPrint(thefun, *a, **k):
savstdout = sys.stdout
sys.stdout = cStringIO.StringIO()
try:
thefun(*a, **k)
finally:
v = sys.stdout.getvalue()
sys.stdout = savstdout
return v
def main():
a = getPrint(funA)
print(a)
if __name__ == "__main__":
main()上面的代码首先将系统标准输出重定向到一个 StringIO 对象，然后调用 funA 函数，并将 funA 函数执行结果存储在变量中。最后将标准输出重定向回原来的位置。
cStringIO 模块也可以用于捕获函数的标准错误输出。以下是如何使用 cStringIO 模块捕获函数的标准错误输出的示例：
import cStringIO
import sys
def getPrintError(thefun, *a, **k):
savstderror = sys.stderr
sys.stderr = cStringIO.StringIO()
try:
thefun(*a, **k)
finally:
v = sys.stderr.getvalue()
sys.stderr = savstderror
return v
def main():
a = getPrintError(funA)
print(a)
if __name__ == "__main__":
main()上面的代码首先将系统标准错误输出重定向到一个 StringIO 对象，然后调用 funA 函数，并将 funA 函数执行结果存储在变量中。最后将标准错误输出重定向回原来的位置。
我们还可以使用 contextlib 模块来实现这一目标。以下是如何使用 contextlib 模块捕获函数输出的示例：
from contextlib import contextmanager
import StringIO
import sys
@contextmanager
def capture():
old_stdout = sys.stdout
sys.stdout = StringIO.StringIO()
try:
yield sys.stdout
finally:
sys.stdout = old_stdout
def main():
with capture() as c:
funA()
funB()
print('HELLO!')
print(c.getvalue())
if __name__ == "__main__":
main()上面的代码首先定义了一个上下文管理器 capture，该上下文管理器将系统标准输出重定向到一个 StringIO 对象。然后使用 with 语句进入上下文管理器，并在该块中调用要捕获输出的函数。最后将标准输出重定向回原来的位置，并将 StringIO 对象的内容作为字符串返回。
最后，我们还可以使用 sys.stdout 和 StringIO 对象来实现这一目标。以下是如何使用 sys.stdout 和 StringIO 对象捕获函数输出的示例：
import StringIO
import sys
def getPrint(thefun, *a, **k):
old_stdout = sys.stdout
sys.stdout = StringIO.StringIO()
try:
thefun(*a, **k)
finally:
v = sys.stdout.getvalue()
sys.stdout = old_stdout
return v
def main():
a = getPrint(funA)
print(a)
if __name__ == "__main__":
main()上面的代码首先将系统标准输出重定向到一个 StringIO 对象，然后调用 funA 函数，并将 funA 函数执行结果存储在变量中。最后将标准输出重定向回原来的位置。
解释
io.StringIO()创建一个在内存中的文本流，用于捕获输出。redirect_stdout(output_io)将标准输出重定向到output_io。output_io.getvalue()从StringIO对象中获取捕获的内容。
这样，无论是返回值还是 print 输出，都可以灵活捕获并用于后续处理。
原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。
如有侵权，请联系 cloudcommunity@tencent.com 删除。
原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。
如有侵权，请联系 cloudcommunity@tencent.com 删除。
腾讯云开发者
