main.py 文件源码

def smooth(arr, sigma=5):
    filled = arr.copy()
    nans = sp.isnan(arr)    

    filled[nans] = 0
    smoothed = sp.ndimage.gaussian_filter(filled, sigma=sigma, truncate=10)  
    missing_coefs = sp.ndimage.gaussian_filter((~nans).astype(float), sigma=sigma, truncate=10)

    corrected = smoothed/missing_coefs

    return corrected