core.py 文件源码

def remove_background_by_median(ccd, plots=False):
    """Remove Background of a ccd spectrum image

    Notes:
        This function works well for images without strong sky lines. Or for
        targets embedded in extended sources.

    Args:
        ccd (object): A ccdproc.CCDData instance.

    Returns:
        ccd (object): The modified ccdproc.CCDData instance.

    """
    new_ccd = ccd.copy()
    data = ma.masked_invalid(new_ccd.data)
    # x, y = ccd.data.shape
    median = ma.median(data, axis=0)

    data -= median
    data.set_fill_value(-np.inf)
    new_ccd.data = data.filled()

    # ccd.write('/user/simon/dummy_{:d}.fits'.format(g), clobber=True)
    return new_ccd