我正在编写一段代码来模拟原子在电场和磁场中的相互作用。这部分要求我在给定高度生成相互作用势的3D图。生成这些图的完整代码很长并且分为几个模块，但是相关的绘图部分是:#Functiontoplot'PlotValues'ataheight'z'defPlot_AtHeight(self,PlotValues,z=500,ReturnFig=False,ShowTime=False):#Callsouttotherelevantfunctiontocalculatethevaluesandreturn#theseasanarrayPlotArray=self.Get_AtHeight(Pl