如何在Python中获取与print()输出效果一致的字符串变量

嘿，这个场景我太熟悉了！我们都知道print()会自动解析字符串里的控制字符（比如\r回车符），但直接查看变量的时候显示的是原始的转义内容。要把print()处理后的结果存成变量，有两种实用的方法：

方法一：捕获print()的输出到变量 #

最简单的方式就是把print()的输出“截胡”存到变量里，这可以借助io.StringIO来实现，相当于模拟一个内存中的文件来接收输出：

import io

mystr = '--More-- \r         \rSerial-Number: 12345'

# 创建StringIO对象用来捕获输出
output_buffer = io.StringIO()
# 把字符串打印到buffer里，end=''避免额外添加换行符
print(mystr, file=output_buffer, end='')
# 从buffer里取出处理后的字符串
newStr = output_buffer.getvalue()

# 验证结果
print(newStr)  # 输出：Serial-Number: 12345
print(repr(newStr))  # 查看变量内容：'Serial-Number: 12345'

这种方法完全复刻了print()的处理逻辑，不管字符串里有多少复杂的控制字符，都能得到和print()完全一致的结果，省心又靠谱。

方法二：手动处理\r控制字符 #

如果你想搞清楚底层逻辑，也可以手动模拟终端处理\r的行为——\r会把光标移到行首，后面的字符会覆盖掉前面的内容。我们可以写个小函数来实现这个逻辑：

mystr = '--More-- \r         \rSerial-Number: 12345'

def process_carriage_return(s):
    current_line = []
    for char in s:
        if char == '\r':
            # 遇到回车符就清空当前行，准备被后面的内容覆盖
            current_line = []
        else:
            current_line.append(char)
    return ''.join(current_line)

newStr = process_carriage_return(mystr)
print(newStr)  # 输出：Serial-Number: 12345

这种方法更灵活，如果你需要自定义处理其他控制字符（比如\n），可以很方便地在函数里扩展逻辑。

备注：内容来源于stack exchange，提问作者Dmitry Perets