Python Partial Differential Equation