Python의 stdout 출력을 문자열 변수로 어떻게 리디렉션할 수 있나요?

from cStringIO import StringIO  # Python 2
# or from io import StringIO  # Python 3

import sys

# Save the original stdout object
old_stdout = sys.stdout

# Create a new StringIO object to capture stdout output
mystdout = StringIO()

# Redirect stdout to the new StringIO object
sys.stdout = mystdout

# Execute code that generates stdout output
# ...

# Restore original stdout
sys.stdout = old_stdout

# Access the captured stdout output
output = mystdout.getvalue()