Python 2d Heat Transfer