如何解决plt.savefig保存FITS图像为JPEG时输出空白的问题

问题描述 #

参照Astropy官方FITS图像绘制教程编写处理程序时，需要将绘制完成的FITS图像导出为JPEG格式供后续流程使用，调用matplotlib常规保存接口plt.savefig("output.jpg")后，导出的JPEG文件为完全空白的图像，与窗口中正常显示的FITS内容完全不符。

故障原因 #

空白导出问题通常由两类触发点导致：

• 代码执行顺序错误：将plt.savefig()写在了plt.show()调用之后。matplotlib执行plt.show()时会完成图像渲染并弹出展示窗口，窗口关闭后当前画布的所有资源会被自动回收，后续执行保存操作只能拿到无内容的空白画布。
• 格式参数不匹配：FITS图像绘制通常会绑定WCS天球投影坐标系，且matplotlib默认画布背景为透明通道，而JPEG格式本身不支持透明通道，直接保存会出现图层丢失、内容不显示的异常。

可行修复方案 #

• 调整代码执行顺序：始终将保存操作放在plt.show()之前执行，确保保存动作触发时画布内容未被销毁。标准可运行代码结构参考：

import matplotlib.pyplot as plt
from astropy.visualization import astropy_mpl_style
from astropy.io import fits
from astropy.wcs import WCS

plt.style.use(astropy_mpl_style)
# 读取目标FITS文件
hdu = fits.open("your_target_file.fits")[0]
wcs = WCS(hdu.header)

# 创建画布绘制FITS内容
fig = plt.figure()
ax = plt.subplot(projection=wcs)
ax.imshow(hdu.data, cmap="gray")

# 先执行保存操作
plt.savefig(
    "output.jpg",
    dpi=150,
    bbox_inches="tight",
    facecolor="white"
)
# 最后弹出图像展示窗口
plt.show()

注意：plt.savefig()与plt.show()的顺序不可颠倒，顺序错误是该问题最高频的触发原因。

• 显式调用画布实例保存：不要依赖pyplot的全局画布状态，使用代码中创建的fig对象直接调用保存方法，避免全局状态混乱导致的保存异常，可将原有保存代码替换为：

fig.savefig(
    "output.jpg",
    dpi=150,
    bbox_inches="tight",
    facecolor="white"
)

关键参数说明：

• dpi=150：设置导出图像的分辨率，可根据实际需求调整数值

• bbox_inches="tight"：自动裁剪画布边缘多余白边，保证坐标轴、色条、标注等元素完整显示

• facecolor="white"：强制设置画布为白色不透明背景，解决JPEG不支持透明通道导致的内容丢失问题

• 兼容兜底方案：如果上述方法仍存在显示异常，可以先导出为无损PNG格式，再通过Pillow库转换为JPEG，规避matplotlib直接导出JPEG的格式兼容问题：

from PIL import Image

# 先导出为无损PNG
fig.savefig("temp_export.png", dpi=150, bbox_inches="tight")
# 转换为RGB模式后存为JPEG
with Image.open("temp_export.png") as im:
    im.convert("RGB").save("output.jpg", quality=95)

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