如何将FITS文件转换为numpy数组

Question

我试着训练一个AI算法来确定星系的光度红移，为了这样做我有一个包含训练数据的FITS文件。我需要将这个FITS文件转换成一种可以在Python中轻松操作的格式，特别是一个numpy数组。我已经尝试过使用astropy，并遵循了以下youtube视频：

https://www.youtube.com/watch?v=goH9yXu4jWw

但是，当我试图转换文件，然后检查数据类型时，它仍然是FITS文件，而不是numpy数组。如果有人能帮忙的话，我会非常感激的！

import astropy.io
from astropy.io import fits


truth_north = fits.open('dr9_pz_truth_north.fits')

data = truth_north[1].data  

当我打印数据类型时，它会给出astropy.io.fits.fitsrec.FITS_rec

有人告诉我，FITS_rec类的行为是一个numpy数组，但是我必须实际将文件转换为numpy数组。

注:我已经把这个问题张贴在物理栈交换上了，但是我的问题没有得到真正的回答。

谢谢!

Answer 1

您的意思是要去掉所有元数据，只保留表中的值吗？这样的话，这就是你要找的吗？

data = np.array(truth_north[1].data)