Несколько таблиц сохранены в одном файле.fits

Question

У меня есть несколько таблиц, созданных с astropy.table.Table, например:

 from astropy.table import Table
 import numpy as np
 #table 1
 ta=Table()
 ta["test1"]=np.arange(0,100.)
 #table 2
 tb=Table()
 tb["test2"]=np.arange(0,100.)

Я могу сохранить их индивидуально .fits файлы, использующие

ta.write('table1.fits')
tb.write('table2.fits')

Но я хотел бы, чтобы они были сохранены в том же .fits файл, каждый из которых отличается hdu, Как я могу это сделать?

Answer 1

Вот пример того, как это сделать здесь. Итак, вы можете сделать что-то вроде следующего:

import numpy as np
from astropy.io import fits

ta = Table()
ta['test1'] = np.arange(0, 100.)
col1 = fits.Column(name=ta.colnames[0], format='E', array=ta)

tb = Table()
tb['test2'] = np.arange(0, 100.)
col2 = fits.Column(name=tb.colnames[0], format='E', array=tb)

cols = fits.ColDefs([col1, col2])

hdu = fits.BinTableHDU.from_columns(cols)
hdu.writeto('table.fits')

который имеет только одну двоичную таблицу HDU, но с двумя столбцами. Кроме того, чтобы добавить их в качестве отдельных HDU, вы можете сделать что-то вроде

ta = Table()
ta['test1'] = np.arange(0, 100.)
col1 = fits.Column(name=ta.colnames[0], format='E', array=ta)

hdu1 = fits.BinTableHDU.from_columns(fits.ColDefs([col1]))

tb = Table()
tb['test2'] = np.arange(0, 100.)
col2 = fits.Column(name=tb.colnames[0], format='E', array=tb)

hdu2 = fits.BinTableHDU.from_columns(fits.ColDefs([col2]))

# create a header
hdr = fits.Header()
hdr['Author'] = 'Me'
primary_hdu = fits.PrimaryHDU(header=hdr)

# put all the HDUs together
hdul = fits.HDUList([primary_hdu, hdu1, hdu2])

# write it out
hdul.writeto('table.fits')

Answer 2

Есть служебная функция astropy.io.fits.table_to_hdu.

Чтобы продолжить ваш пример, если у вас есть два табличных объекта ta а также tb:

from astropy.io import fits
hdu_list = fits.HDUList([
    fits.PrimaryHDU(),
    fits.table_to_hdu(ta),
    fits.table_to_hdu(tb), 
])
hdu_list.writeto('tables.fits')