如何修复BytesIO保存Matplotlib图为numpy数组时返回空白的问题

返回全255纯白数组由两个疏漏导致：

• 调用fig.savefig(..., format="raw")前未强制触发画布全量渲染：非交互模式下散点、坐标轴标签等元素不会自动绘制到画布缓冲区，读取到的是初始状态的纯白画布数据。
• 未对raw格式输出的一维字节流做维度重塑：raw格式仅输出无结构的像素字节流，即使渲染正确，直接读取的一维数组也无法作为有效图像使用。

方案1：保留原有BytesIO+raw格式逻辑 #

只需要补全强制渲染、维度重塑两步即可：

import lightkurve
import matplotlib.pyplot as plt
import numpy as np
import io

def download(search):
    lc = search.download()
    if lc is not None:  
        fig, ax = plt.subplots()
        ax.scatter(lc.time.value.tolist(), lc.flux.value.tolist(), color='k')
        ax.autoscale()
        ax.set_xlabel('Time (BTJD)')
        ax.set_ylabel('Flux')
        
        # 强制画布完成所有元素渲染，必须在savefig前调用
        fig.canvas.draw()

        io_buf = io.BytesIO()
        fig.savefig(io_buf, format="raw")
        io_buf.seek(0)
        
        # 获取画布实际宽高，按RGBA4通道规则重塑数组维度
        width, height = fig.canvas.get_width_height()
        img_arr = np.frombuffer(io_buf.getvalue(), dtype=np.uint8).reshape((height, width, 4))
        io_buf.close()
        
        plt.close(fig) # 关闭画布避免内存泄漏
        # 如需3通道RGB格式，可追加：img_arr = img_arr[..., :3]
        return img_arr

方案2：更简洁的无中转方案 #

不需要BytesIO做中转，直接从渲染完成的画布缓冲区提取数组，出错概率更低：

import lightkurve
import matplotlib.pyplot as plt
import numpy as np

def download(search):
    lc = search.download()
    if lc is not None:  
        fig, ax = plt.subplots()
        ax.scatter(lc.time.value.tolist(), lc.flux.value.tolist(), color='k')
        ax.autoscale()
        ax.set_xlabel('Time (BTJD)')
        ax.set_ylabel('Flux')
        
        fig.canvas.draw()
        # 直接读取画布RGBA缓冲区数据
        img_arr = np.array(fig.canvas.buffer_rgba(), dtype=np.uint8)
        
        plt.close(fig)
        # 如需3通道RGB格式，可追加：img_arr = img_arr[..., :3]
        return img_arr

• 批量生成图像时建议去掉调试用的fig.show()：部分GUI后端的show()是异步渲染逻辑，调用后立刻读取缓冲区可能拿到未渲染完成的空白数据，读数组操作不需要依赖画布显示。
• 之前尝试换用plt全局操作、调用lc.plot(ax=ax)仍然失败，本质都是没有执行fig.canvas.draw()强制渲染，和绘图调用方式无关。
• 如果在交互环境（如Jupyter）中运行，确保使用默认的inline后端即可，不要切换到未实现缓冲区读取接口的定制后端。

内容的提问来源于stack exchange，提问作者Sophie Frankel