python version of Lagrangian Advection and DIffusion Model