Question

假设我有一个带有一个扩展名的拟合文件，数据由两列的表组成，每列有100个元素

data = pyfits.open('path2myfile')[1].data
head = pyfits.open('path2myfile')[1].header
print data['field1'] # print an array with 100 elements
print data['field2'] # print another array with 100 elements

现在我想在我的表中添加一个新列，比如数据['field3']，这是另一个包含100个元素的数组。

我该怎么做？

Answer 1

正如Iguananaut所说，答案可以在这里找到：http://pyfits.readthedocs.org/en/latest/users_guide/users_table.html#merging-tables 但只是为了回答这个问题：

cols = [] 
cols.append(
    pyfits.Column(name='field3', format='D', array= arrayContainingTheData)
    )
orig_cols = data.columns
new_cols = pyfits.ColDefs(cols)
hdu = pyfits.BinTableHDU.from_columns(orig_cols + new_cols)
hdu.writeto('newtable.fits')

Answer 2

@ Cehem的回答是正确的。但我只想补充一点，Astropy有一个更好的通用Table接口，您可能会发现它更容易使用（在其他用例中插入列）。

Astropy在astropy.io.fits模块中包含了PyFITS。但是，由于您还可以访问更好的Table接口，因此您可以将大多数 FITS表读入Astropy表类中，如下所示：

>>> from astropy.table import Table
>>> table = Table.read('path/to/fits_file.fits')

就是这样。您还可以将表写回FITS文件。当然，这目前不支持所有类型的FITS表，但它适用于大多数 - 以及将来所有。

使用PyFITS将列添加到FITS表

2 个答案: