python - 保存 2 个带有 3 个标题 astropy 或 pyfits 的文件

Question

我正在尝试制作一个包含两个图像的适合文件，并且我想给它们两个标题。他们两个都应该有一个基本的标题。到目前为止我发现的是这个，但我不知道如何正确添加标题。变量是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 的astropy 文档中，您可以将标头作为关键字传递给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)