Solving Heat Equation In Python Which Library