以字符串形式获取函数的输出

Question

我有一个打印功能。如何将其输出为字符串？注意：我无法更改功能代码。def f():    print('hello world')# do somethingassert x == 'hello world\n'我需要这个，因为火花数据框中有方法explain即打印约执行计划的信息。但我需要这个信息作为我的程序中的字符串。>>> df.explain()== Physical Plan ==Scan ExistingRDD[age#0,name#1]

Answer 1

def f():

    print('hello world')

import io

from contextlib import redirect_stdout

with io.StringIO() as buf, redirect_stdout(buf):

    f()

    x = buf.getvalue()

assert x == 'hello world\n'