base.py 文件源码

def read_fits(filename='long.fits'):
    """
    Read an sample fits file and return only image part.

    Parameters
    ----------
    filename : str
        Fits filename.

    Returns
    -------
    data : numpy.ndarray
        Fits image data.
    """

    module_path = dirname(__file__)
    file_path = join(module_path, 'samples', filename)

    hdulist = fits.open(file_path)

    data = hdulist[0].data

    hdulist.close()

    return data