在python中使用有限差分

Question

我尝试使用Python和Numpy来使用有限差分法来求解一个基本方程。代码给了我正确的第一个值，也就是说它给了我一个[1]；但是，之后的其他值都是零？ 我不知道我做错了什么，因为它显然适用于第一个值，所以我该如何修复它？ 任何想法都会很有帮助。在

^{1}$

Answer 1

函数在第一次迭代后返回，因为return语句在for循环中。应该删除return语句，这样循环就不会过早终止：

for i in range (0, n):
    x[i+1] = x[i] + (H_0 * ((omega_m)**(1./2.)) * ((x[i])**(-1./2.)) * dt)
return x, t