buf = io.BytesIO()plt.savefig(buf, format='png')buf.seek(0)plt.close()

io.StringIO() 和 io.BytesIO() 是 Python 标准库 io 模块中的两个类，分别用于创建内存中的文本流和二进制流。它们常常用于单元测试、文件操作的模拟等场景。

基础概念

• io.StringIO(): 创建一个内存中的文本流，可以像文件一样进行读写操作，但数据存储在内存中，而不是磁盘上。
• io.BytesIO(): 创建一个内存中的二进制流，同样可以像文件一样进行读写操作，但数据是以字节形式存储。

相关优势

• 内存效率：相比于磁盘文件，内存流操作更快，因为不需要进行磁盘 I/O。
• 测试友好：在单元测试中，可以使用内存流来模拟文件操作，避免了对真实文件的依赖。
• 灵活性：内存流可以在程序运行时动态创建和销毁，提供了更大的灵活性。

类型与应用场景

• io.StringIO(): 适用于需要处理文本数据的场景，如读取配置文件、生成日志等。
• io.BytesIO(): 适用于需要处理二进制数据的场景，如图片处理、数据加密等。