如何在编辑.fits文件后保存它且不丢失头部信息？

Question

我有一个.fits文件，我正在编辑，虽然我可以编辑和打印它很好，当我保存它时，我会丢失标题中的所有信息。下面是我的代码：

import numpy as np
from astropy.io import fits
import matplotlib.pyplot as plt

im = fits.getdata('myimage.fits')

ID = 403

newim = np.copy(myim)

newim[newim == ID] = 0
newim[newim == 0 ] = -99
newim[newim > -99] = 0
newim[newim == -99] = 1

所有这些都很好，但当我保存新的fits文件时，我遇到了一些问题。我尝试了以下操作，但再次它没有保存标题中包含的信息：

hdu = fits.PrimaryHDU(newim)
hdu.writeto('new image.fits')

有没有办法保存我的图像而不丢失标题中的所有信息

Answer 1

我设法从一位朋友那里得到了答案：

import numpy as np
from astropy.io import fits
import matplotlib.pyplot as plt

im = fits.getdata('myimage.fits')
403header = fits.getheader('myimage.fits')

ID = 403

newim = np.copy(myim)

newim[newim == ID] = 0
newim[newim == 0 ] = -99
newim[newim > -99] = 0
newim[newim == -99] = 1

hdu = fits.PrimaryHDU(newim, header=403header)
hdu.writeto('new_image.fits')