Python热图方程

我需要帮助尝试将方程式转换为Python，以便可以从中生成热图。它描述了以x轴为中心，左边缘在y轴上的板的温度分布。平板在x轴上宽x =宽，y = b / 2高，在x轴下高y = -b / 2。平板的顶部和底部保持120度，侧面保持20度。下面是查看方程式图像的链接。

如何将其转化为Python可以识别的代码？

from mpmath import *
T = 20 + nsum(lambda n: (-1)**n * (400/ (n * pi)) * (cosh((n*pi*y)/a) / cosh((n*pi*b)/2*a)) * sin((n*pi*x)/a),[1,inf])