SciPy: 2D diffusion