如何使用FitsIO更新FITS文件头?

时间:2019-04-18 14:40:30

标签: python python-3.x fits

我当前正在尝试修改FITS文件的标题。更具体地说,我想为其添加新的关键字。我使用的是FitsIO for python,版本为0.9.11。

在进行一些数据处理期间,我的程序创建了新的FITS文件并更新了它们的内容和标题。我曾经使用过Astropy,但是现在这已不再是一种选择,因此我只能使用FitsIO,这对我来说是个挑战。

内容更新/覆盖非常简单,但是我在修改标题时遇到了麻烦。为了隔离我的问题并找到正确的过程(该文档不是很有帮助),我创建了一个小程序,仅测试如何编写标头。

作为输入数据,我使用包含以下内容的FITSHDR对象:

update_data = fitsio.FITSHDR([
"IVAL    =                   36 / integer value           ",
"SHORTS  = 'hello w'                                      ",
"DBL     =                 1.25                           ",
"CDELT1  =                 0.00333333333333               "])

这是经过测试的功能:

def write_hdu_1(data, fits_file):
    with fitsio.FITS(fits_file, 'rw', clobber=True) as fits:
        fits[extension].write_keys(data)

结果:删除END卡,损坏HDU。

def write_hdu_2(data, fits_file):
   with fitsio.FITS(fits_file, 'rw', clobber=True) as fits:
       for key in data.keys():
           fits[extension].write_key(key, data.get(key))

结果:删除END卡,损坏HDU。

def write_hdu_3(fits_file):
    with fitsio.FITS(fits_file, 'rw', clobber=True) as fits:
        keydict = { 'name':'test', 'value':35, 'comment':'keydict test'}
        fits[-1].write_key(**keydict)

结果

File "test_fitsio.py", line 26, in write_hdu_3
    fits[-1].write_key(**keydict)
TypeError: write_key() got an unexpected keyword argument 'name'

无奈之下,我终于手动添加了“ END”卡,就像这样:

def write_hdu(self, header_content):
    self.fits[extension].write_keys(header_content)
    self.fits[extension].write_key('END', '')

但是,然后,在END卡上填充了空格,这是我正在使用的其他库无法正确识别的。

那么,您对使用python FitsIO添加/更新标头关键字有什么后见之明,而又不会在此过程中丢失“ END”卡?

谢谢!

1 个答案:

答案 0 :(得分:1)

在打开文件时删除clobber选项,然后将使用标头版本。使用clobber选项可以在打开任何现有文件之前将其删除,请参见docstringcode的实现。

在我这方面,这就像Python3中fitsio 0.9.11的魅力:

with fitsio.FITS('test.fits', 'rw') as ff:
    ff[-1].write_key('KEY', 'key value', 'my comment')

另一方面,如果使用clobber选项,则在现有文件上使用相同代码在给定的HDU标头中附加关键字,从而从磁盘上删除文件。

$ du -sh test.fits                                                                                           
308K    test.fits
$ python3 -c "import fitsio; ff = fitsio.FITS('test.fits', 'rw', clobber=True); ff[-1].write_key('KEY', 'key value', 'my comment')"
$ du -sh test.fits                                                                                  
4.0K    test.fits
$ cat test.fits
KEY     = 'key value'          / my comment                                     END                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                        

这是一个很奇怪的结果,但是fitsio附加了END关键字,但是在一个空文件上,如果不是有效的FITS文件,则为空文件。