您的代码有几个问题,如果您更仔细地阅读文档字符串odeint
,其中大部分可能会自己解决。
为了帮助您入门,以下是求解标量微分方程的简单示例odeint
。我将使用一个非常简单的方程式,而不是尝试理解(并可能调试)您的函数。我将求解方程 dy/dt = a * y,初始条件 y(0) = 100。一旦你有这个例子工作,你可以修改fun
来解决你的问题。
import numpy as np
from scipy.integrate import odeint
import matplotlib.pyplot as plt
def fun(y, t, a):
"""Define the right-hand side of equation dy/dt = a*y"""
f = a * y
return f
# Initial condition
y0 = 100.0
# Times at which the solution is to be computed.
t = np.linspace(0, 1, 51)
# Parameter value to use in `fun`.
a = -2.5
# Solve the equation.
y = odeint(fun, y0, t, args=(a,))
# Plot the solution. `odeint` is generally used to solve a system
# of equations, so it returns an array with shape (len(t), len(y0)).
# In this case, len(y0) is 1, so y[:,0] gives us the solution.
plt.plot(t, y[:,0])
plt.xlabel('t')
plt.ylabel('y')
plt.show()
这是情节:
更复杂的使用示例odeint
可以在SciPy Cookbook中找到(向下滚动到标有“普通微分方程”的项目符号)。