使用with语句的BytesIO将Matplotlib图转PIL图片失败的解决方法

数据老陈

2026-6-13

问题：如何在with语句中直接生成可正常显示的PIL图片？

背景

我正在开发一个函数，流程为：接收包含数据的字典 → 基于数据使用非交互式Matplotlib绘图 → 将渲染后的图转为PIL图片并保存回字典 → 返回该字典（后续会转换为pandas.DataFrame，用于快速查看其中的图片）。

遇到的异常

使用BytesIO()缓冲区时性能表现良好，但通过with语句管理缓冲区时，生成的图片无法正常显示：

调用IPython.display.display(image)仅返回句柄，无法显示图片
调用image.show()会抛出I/O错误
该问题已在to_PIL_direct()函数中复现。

现有规避方案（存在缺陷）

目前有两种可行但不够理想的解决方法：

方案1：将缓冲区内容转为字节，再通过字节流读取生成PIL图片（对应to_PIL_bytes()函数），但字节转换属于冗余操作，会降低代码性能
方案2：完全不使用with语句管理缓冲区（对应to_PIL_noclose()函数），但无法主动关闭缓冲区，存在资源泄漏风险

需求

我需要实现一个函数，既能达到上述两种方案的预期输出效果，又能高效地关闭并清理缓冲区，避免冗余操作和资源泄漏问题。

参考实现思路

最初的实现思路参考了Stack Overflow上dizcza回答中的plot2()函数：

import io
import matplotlib
matplotlib.use('agg')  # 关闭交互式后端
import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots()
ax.plot(range(10))


def plot1():
    ...


def plot2():
    with io.BytesIO() as buff:
        fig.savefig(buff, format='png')
        buff.seek(0)
        im = plt.imread(buff)

def plot3():
    ...