我的毕业论文是用python做数值分析。因为本人没有编程基础，就让AI帮写的。

遇到的问题是：AI写出来的代码是个半成品，没有插入物理公式，所以运行出来的图是错的，不符合物理规律（当然我也不太清楚图片错误的具体原因，可能输入了物理公式也运行不出来）。需求是：在这个代码的基础上进行修改，帮我输入已知的物理公式，跑出来正确的图。
下面附件中两个图都是错的，还需要输入已知的物理公式。