SciPy: 1D diffusion