用Python制作fits文件

前面的博客已经讲了fits文件是什么，以及如何读fits文件，当你完成一项工作的时候，你可能也会想制作一个自己的fits文件来发布自己的数据，本篇博客就讲一下如何制作自己的fits文件

通常我们是使用pandas进行数据分析，那么我们最终的数据就会是DataFrame格式，本文就从DataFrame格式的数据出发

1. 第一步我们需要给定fits文件的列名，每一列的格式，以及每一列的单位

from astropy.io import fits
from pandas as pd

column_name = list(data)
format_list = ['K','D','D','D','E','D']
unit_list = ['','deg','deg','mas','mas','mas/yr']

data是一个DataFrame格式的文件，list(data)表示取出data的全部列名组成一个list，fits中每一列的格式和我们通常所见的int, float等不大一样，具体的对应关系如下图所示

常用的就是L（布尔型），K（64位整型），D（双精度浮点数），A（字符串） ，对于字符串需要指定字符串的长度，也就是诸如100A，20A这种表达

2. 第二步生成一个fits文件并保存，代码如下：

c = list(np.zeros(len(list(data))))

for i in range(len(c)):
    c[i] = fits.Column(name=column_name[i], array=np.array(data[column_name[i]]),format=format_list[i],unit=unit_list[i])
    
t = fits.BinTableHDU.from_columns(c)
t.writeto('filename.fits')

注意这里是一列一列的加入的

3. 如果你的数据不是DataFrame格式，道理也是类似的，先赋予列名，格式，单位，然后按照列名一列列加入数据，也就是把fits.Column里面的array换成你对应列的数据即可

4. topcat也可以把其他格式的文件转化为fits文件，先读入文件，然后在保存文件的时候格式选择fits即可，不过topcat保存的fits文件可能会导致Python读取不了，具体的原因我没有搞清楚，但是有时候的确是不行的