保存2个带有3个头文件的拟合文件astropy或pyfits

Question

我正在尝试制作一个带有两个图像的fits文件，并且我想同时给它们一个标题。 两者都应有一个基本标头。 到目前为止，我发现的是这个，但是我不知道如何正确添加标题。 变量为headermain，header1，header2，data1，data2。 我希望它具有这种结构。

Filename: test.fits
No.    Name         Type      Cards   Dimensions   Format
0    PRIMARY     PrimaryHDU     828   ()              
1    name1  ImageHDU        52   (1024, 900)   float32   
2    name2  ImageHDU        52   (1024, 900)   float32 

我目前有这个，但是我找不到如何添加标题。

new_hdul = pyfits.HDUList()
new_hdul.append(pyfits.ImageHDU(data1)
new_hdul.append(pyfits.ImageHDU(data2)

new_hdul.writeto('test.fits', clobber=True)

Answer 1

从ImageHDU的原始文档中，您可以将标头作为关键字传递给ImageHDU ：

from astropy.io import fits

new_hdul = fits.HDUList()
new_hdul.append(fits.PrimaryHDU(header=headermain))
new_hdul.append(fits.ImageHDU(data1, header=header1, name='name1'))
new_hdul.append(fits.ImageHDU(data2, header=header2, name='name2'))

new_hdul.writeto('test.fits', clobber=True)