如何在Pyglet窗口中嵌入Matplotlib绘图 解决Python3兼容问题

你怀疑的位置完全正确，问题的核心是Python3对字符串和字节流做了严格区分：

• io.StringIO仅支持存储Unicode文本数据，而Matplotlib画布输出的原始图像是二进制字节流，写入StringIO会直接抛出类型不匹配错误。
• 原代码中的"是HTML转义后的双引号，实际编写代码时需要替换为普通双引号。

import pyglet
import io
import numpy as np

from matplotlib.figure import Figure
from matplotlib.backends.backend_agg import FigureCanvasAgg


def render_figure(fig):
    w, h = fig.get_size_inches()
    dpi_res = fig.get_dpi()
    w, h = int(np.ceil(w * dpi_res)), int(np.ceil(h * dpi_res))

    canvas = FigureCanvasAgg(fig)
    pic_data = io.BytesIO()  # 替换为BytesIO存储二进制流
    canvas.print_raw(pic_data, dpi=dpi_res)
    return pyglet.image.ImageData(w, h, "RGBA", pic_data.getvalue(), -4 * w)


def draw_figure(fig):
    X = np.linspace(-6, 6, 1024)
    Y = np.sinc(X)

    ax = fig.add_subplot(111)
    ax.plot(X, Y, lw=2, color="k")


# 可以把fullscreen=True改为width=800, height=600方便测试
window = pyglet.window.Window(fullscreen=True)
dpi_res = min(window.width, window.height) / 10
fig = Figure((window.width / dpi_res, window.height / dpi_res), dpi=dpi_res)

draw_figure(fig)
image = render_figure(fig)


@window.event
def on_draw():
    window.clear()
    image.blit(0, 0)


pyglet.app.run()

执行以下命令安装所需库即可：
pip install pyglet numpy matplotlib

内容的提问来源于stack exchange，提问作者Philippe Maisonneuve