raw 图像转成fits格式图像

import numpy as np
from astropy.io import fits

alist = [line.rstrip()for line in open('raw.dat')]
for file in alist:
    raw = file
    img=np.fromfile(raw,dtype=np.uint16,offset=1024)
    b=np.reshape(img,(4096,4096))
    fit = file[0:-4]+'.fits'
    fits.writeto(fit,b)