def blur_grid(grid): filtered = gaussian_filter(grid, sigma=1) filtered[(grid > 0.45) & (grid < 0.55)] = grid[(grid > 0.45) & (grid < 0.55)] return filtered