Python中StringIO()的常见用法总结

StringIO()是Python中的一个类，用于在内存中操作字符串。它提供了一个类似于文件对象的接口，可以对字符串进行读写操作。下面是StringIO()的常见用法总结：

1. 写入字符串：

使用StringIO()对象的write()方法可以很方便地向对象中写入字符串。

from io import StringIO

s = StringIO()
s.write("Hello, World!")

2. 读取字符串：

使用StringIO()对象的getvalue()方法可以获取对象中的全部内容，返回一个字符串。

from io import StringIO

s = StringIO("Hello, World!")
print(s.getvalue())

3. 读取部分内容：

使用StringIO()对象的read()方法可以读取指定长度的内容，参数是要读取的长度。

from io import StringIO

s = StringIO("Hello, World!")
print(s.read(5))

4. 将字符串作为文件对象处理：

StringIO()对象可以直接传给其它需要文件对象的函数或方法。例如，可以将StringIO()对象传给json.load()函数，实现将字符串转换为json对象。

from io import StringIO
import json

s = StringIO('{"name": "John", "age": 30}')
data = json.load(s)
print(data)

5. 重置对象：

使用StringIO()对象的seek()方法可以将读写位置复位到起始位置。这样就可以重新读取全部内容或者覆盖原内容。

from io import StringIO

s = StringIO("Hello, World!")
print(s.getvalue())  # 输出"Hello, World!"

s.seek(0)  # 将读写位置复位到起始位置
s.write("Goodbye, World!")  # 将原内容覆盖
print(s.getvalue())  # 输出"Goodbye, World!"

6. 与标准输入输出流的交互：

StringIO()对象作为内存中的输入或输出流，可以与标准输入输出流进行交互。例如，可以将标准输出流的内容重定向到StringIO()对象，比如用于测试函数的输出结果。

from io import StringIO
import sys

s = StringIO()
sys.stdout = s
print("Hello, World!")
output = s.getvalue()
sys.stdout = sys.__stdout__

print(output)

上述是StringIO()的常见用法总结及使用例子。StringIO()是一个非常方便的工具，在需要在内存中操作字符串的场景中经常会用到。