python输出语句print i、j、k_python：在exec语句中获取打印输出

I want to get the output of an exec(...) Here is my code:

code = """

i = [0,1,2]

for j in i :

print j

"""

result = exec(code)

How could I get the things that print outputed?

How can I get something like:

0

1

2

Regards and thanks.

解决方案

I had the same idea as Frédéric, but i wrote a context manager to handle replacing stdout:

import sys

from io import StringIO

import contextlib

@contextlib.contextmanager

def stdoutIO(stdout=None):

old = sys.stdout

if stdout is None:

stdout = StringIO()

sys.stdout = stdout

yield stdout

sys.stdout = old

code = """

i = [0,1,2]

for j in i :

print j

"""

with stdoutIO() as s:

exec(code)

print("out:", s.getvalue())