Question

我正在尝试使用Python和Numpy来使用有限差分法求解基本方程。代码给了我正确的第一个值，即它给了我一个[1];但是，之后的每个其他值都只是零？我不知道我做错了什么，因为它显然适用于第一个值，所以我该如何解决这个问题呢？任何想法都会非常有用。

from numpy import *
import numpy as np
import matplotlib.pyplot as plt
import scipy as sp
from scipy.integrate import odeint

def solver(omega_m, dt):
    #t_0, H_0, a_0, dt, n, T; always the same; omega's change
    t_0 = 0.0004
    a_0 = 0.001
    H_0 = 1./13.7

    T = 13.7
    dt = float(dt)
    n = int(round((T - t_0)/dt))
    x = zeros(n+1)
    t = linspace(t_0, T, n+1)

    x[0] = a_0

    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

a, t = solver(omega_m =1, dt=0.001)
print a, t

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

在python中使用有限差分

1 个答案: