您好我正在编写一个脚本,它将使用Runge Kutta方法解决并绘制ODE。我希望脚本使用不同的功能,以便稍后我可以扩展它。如果我用函数定义编写它就可以正常工作,但是使用def()它不会打开绘图窗口或打印结果。比你!
from numpy import *
import matplotlib.pyplot as plt
#H=p^2/2-cosq
#p=dp=-dH/dq
#q=dq=dH/dp
t = 0
h = 0.5
pfa = [] #Create arrays that will hold pf,qf values
qfa = []
while t < 10:
q = 1*t
p = -sin(q*t)
p1 = p
q1 = q
p2 = p + h/2*q1
q2 = q + h/2*p1
p3 = p+ h/2*q2
q3 = q+ h/2*p2
p4 = p+ h/2*q3
q4 = q+ h/2*p4
pf = (p +(h/6.0)*(p1+2*p2+3*p3+p4))
qf = (q +(h/6.0)*(q1+2*q2+3*q3+q4))
pfa.append(pf) #append arrays
qfa.append(qf)
t += h #increase time step
print("test")
plt.plot(pfa,qfa)
print("test1")
plt.show()
print("tes2t")
答案 0 :(得分:1)
如果你声明了一个函数,你需要在某个时候调用它,例如:
def rk(p,q,h):
pass # your code here
if __name__ == '__main__':
rk(1,2,1)
将函数调用放在if __name__ == '__main__'
块中将确保仅在直接运行脚本时调用该函数,而不是在从其他脚本导入时调用该函数。 (如果您有兴趣,请点击此处:What does if __name__ == "__main__": do?)
这是一个更好的选择;避免硬编码fn args(你的真实代码应该对意外的命令行输入有一些错误处理):
def rk(p,q,h):
pass # your code here
if __name__ == '__main__':
import argparse
the_parser = argparse.ArgumentParser()
the_parser.add_argument('integers', type=int, nargs=3)
args = the_parser.parse_args()
p,q,h = args.integers
rk(p,q,h)