我正在尝试使用Vertel算法求解微分方程,但我无法摆脱这个错误。有什么建议吗?
def Vertel(dt, x0, v0, t, tstop, pravastrana, M=1):
i = 0
x = x0
v = v0
k = 1 # Spring constant
m = 1
while t <= tstop:
a = -(k/m)* x[i]
if i == 0:
v_next = v[i] + a* dt
x_next = x[i] + v_next* dt
else:
x_next = 2* x[i] - x[i-1] + a* dt** dt
x.append(x_next)
t.append(t[i] + dt)
i = i + 1
t = t + dt
return(x, t)
print(*Vertel(0.1 ,1 ,1 , 0, 10, pravastrana_1, ))
在我定义a
的行上,我得到错误消息:
> 'int' object is not subscriptable
感谢任何帮助,谢谢
您想将x
和v
初始化为列表,
x = [x0]
v = [v0]
为了获得Verlet算法的正确顺序,您还可以考虑将第一步初始化为高一个顺序,如
v_next = v[i] + 0.5*a* dt
x_next = x[i] + v_next* dt
因为这给出了二次项的正确泰勒展开式。请注意,现在半步位置的速度是正确的,因为它将用于Leapfrog-Verlet变体。
会有一些问题被第一个错误掩盖,比如t
同时作为标量变量和列表的实际双重处理,甚至可能更多。