python - 使用 Astropy 将 3d Numpy 数组写入 FITS 文件

Question

我有一个代表 256x256 图像的 3D NumPy 数组(即 (10, 256, 256))。我想使用 astropy.io.fits 将此数组写入 FITS 文件，以便我可以使用 ds9 -mecube 打开该文件并在帧中移动。我的尝试如下所示

export_array = numpy.array(images) #Create an array from a list of images
print export_array.shape ## (10, 256, 256)

hdu = fits.PrimaryHDU(export_array)
hdulist = fits.HDUList([hdu])
hdulist.writeto(out_file_name)
hdulist.close()

这将为我提供一个 FITS 文件，该文件实际上包含 3D 数组。但是，如果我使用 ds9 -mecube 打开，我只能看到第一张图像。无论如何，是否可以使用 astropy.io.fits 创建具有此功能的 FITS 文件？或者 ds9 是否有一些我缺少的功能？

Answer 1

我不使用 ds9，但显然 -mecube 选项意味着“多扩展立方体”。文档说“将多扩展 FITS 文件加载为数据立方体。您只需将单个数组编写为数据立方体。要将其编写为多扩展 FITS，您可能会执行以下操作:

hdul = fits.HDUList()
hdul.append(fits.PrimaryHDU())

for img in export_array:
    hdul.append(fits.ImageHDU(data=img))

hdul.writeto('output.fits')

(您不需要调用 hdul.close()——只有当 HDUList 是从现有文件加载并且您想要关闭时，它才会执行任何操作底层文件对象；它对在内存中从头开始创建的 HDUList 没有影响)。

我不知道 ds9 期望将多扩展名 FITS 文件加载为数据立方体 - 这不是任何特定的 FITS 约定，文档也不清楚。但大概是这样的。

话虽如此，根据 ds9 文档，您根本不需要使用它。如果您不使用 -mecube 选项，它只会将主 HDU 中的 3D 数组读取为数据立方体。