median.py 文件源码

def median_baseline(intensities, window_size=501):
    '''Perform median filtering baseline removal.
    Window should be wider than FWHM of the peaks.
    "A Model-free Algorithm for the Removal of Baseline Artifacts" Friedrichs 1995
    '''
    # Ensure the window size is odd
    if window_size % 2 == 0:
        window_size += 1
    # Enable batch mode
    if intensities.ndim == 2:
        window_size = (1, window_size)
    return medfilt(intensities, window_size)