python - 如何使用 FitsIO (python) 创建空的 FITS 图像？

Question

我想通过给出输入尺寸来创建一个空的 FITS 图像。我这样做是因为我迭代图像的内容并修改它，所以我需要首先使用空图像初始化文件。

使用 astropy，这很容易，但我正在将库切换到 FitsIO，但我无法将此代码转换为实际工作的代码。我一直在查找 FitsIO 的 github 项目，发现了一个名为 write_empty_hdu 的 API，但我显然误用了它。

这是我的功能:

 def create_image(self, x_size, y_size, header=None):
    """!
    @brief Create a FITS file containing an empty image.
    @param x_size  The number of pixels on the x axis.
    @param y_size  The number of pixels on the y axis.
    """
     if header is not None:
         self.fits.write_empty_hdu(dims=[x_size, y_size], header=header, clobber=True)
    else:
         self.fits.write_empty_hdu(dims=[x_size, y_size], clobber=True) 

结果如下:

ERROR : 'FITS' object has no attribute 'write_empty_hdu'

Traceback (most recent call last):

File "/home/user/Work/Projects/DET_CL_PZWAV/DET_CL_PZWav/python/DET_CL_PZWav/LE3_DET_CL_PZWav.py", line 159, in mainMethod pixel_grid.initialize_map()

File "/home/user/Work/Projects/DET_CL_PZWAV/DET_CL_PZWav/python/DET_CL_PZWav/Grid.py", line 232, in initialize_map

self.wavelet_map.create_image(self._pixel_number['Ny'], self._pixel_number['Nx'])

File "/home/user/Work/Projects/DET_CL_PZWAV/DET_CL_PZWav/python/DET_CL_PZWav/PixelSmoothedImage.py", line 57, in create_image super().create_image(x_size, y_size)

File "/home/user/Work/Projects/DET_CL_PZWAV/DET_CL_PZWav/python/DET_CL_PZWav/FITSImage.py", line 81, in create_image self.fits.write_empty_hdu(dims=[x_size, y_size], clobber=True)

AttributeError: 'FITS' object has no attribute 'write_empty_hdu'

你知道我如何使用 FitsIO 编写这个图像创作吗？

谢谢!

Answer 1

总的来说，Python内置的内省(introspection)和帮助功能非常强大。例如:

>>> import fitsio
>>> dir(fitsio)
['ASCII_TBL', 'BINARY_TBL', 'FITS', 'FITSCard', 'FITSHDR', 'FITSRecord', 'FITSRuntimeWarning', 'IMAGE_HDU', 'READONLY', 'READWRITE', '__builtins__', '__cached__', '__doc__', '__file__', '__loader__', '__name__', '__package__', '__path__', '__spec__', '__version__', '_fitsio_wrap', 'cfitsio_version', 'fitslib', 'read', 'read_header', 'read_scamp_head', 'test', 'util', 'write']

正如您所看到的，没有 write_empty_hdu，但有一个 write，看起来很有希望。所以现在:

>>>help(fitsio.write)

将向您展示您需要了解的一切。在你的情况下，你可能想要:

fitsio.write('somefile',np.empty(shape=(3,4)),header={'a': '','b': 'a','c': 3},clobber=True)

请注意，numpy.empty 可能会写入任意值 - 因此您可能需要零来确保您不相信数据是真实的。