Two different Python implementations of both one- and two-dimensional diffusion equations are available on my GitHub account. I have also included explanations for each line of code. You can access it by clicking here.