我写了一段代码来在Python上使用四阶Runge-Kutta求解Lotka-Volterra方程,但是由于某种原因它不起作用,因此解决方案是完全错误的。老实说,我不知道我在做什么错。
import numpy
from pylab import plot, show
def rk(f, x, h):
f_1 = f(x)
f_2 = f(x+1./2*h*f_1)
f_3 = f(x+1./2*h*f_2)
f_4 = f(x+h*f_3)
return x+1./6*h*(f_1+2*f_2+2*f_3+f_4)
def lv(x):
alpha = 1.
return numpy.array([alpha*x[0]-x[0]*x[1], x[0]*x[1]-x[1]], float)
a = 0.
b = 10.
m = 100
T = numpy.linspace(a, b, m)
H = (b-a)/m
X = numpy.zeros((m, 2))
X[0, :] = [1., 30.]
for i in range(1, m):
X[i, :] = rk(lv, X[i-1, :], H)
plot(T, X)
show()
非常感谢您的帮助。