【发布时间】:2022-07-22 01:25:03
【问题描述】:
我正在求解一个耦合微分方程组,微分方程中的一个“常数”实际上是一个周期性变化的值:周期的前半部分的值为 1,周期的其余部分的值为为 0,周期为 2pi。
我将该常数的值设置为平方函数(使用平方函数代码和 if else 代码),并使用 odeint 求解微分方程。
import numpy as np
from scipy.integrate import odeint
import matplotlib.pyplot as plt
import math
from scipy import signal
u=0.3
def l(f):
if int(2*t)%2np.pi == 0:
return 1
else:
return -1
def model(theta,t):
j = theta[0]
x = theta[1]
p = theta[2]
dJ_dt = 2*l(f)*(math.sqrt(2))*x*j
dX_dt = p-(k/2)*x
dP_dt = -x-(k/2)*p-2*l(f)*(math.sqrt(2))*j
dtheta_dt = [dJ_dt, dX_dt, dP_dt]
return dtheta_dt
theta_0 = [0.5*math.sqrt(1-u**2), 0, -0.5*u, 0, 0]
t = np.linspace(0,5000,1000)
theta = odeint(model,theta_0,t)
plt.figure(figsize=(25,8))
plt.plot(t, theta[:,0],label='j')
plt.legend(fontsize=15)
plt.xlabel('Time', fontsize= 30)
plt.xticks(fontsize= 20)
plt.ylabel('jx', fontsize= 30)
plt.yticks(fontsize= 20)
plt.show()
但似乎由于常量不是“标量”,代码不可解。
我也参考过这个帖子:Plotting solved ordinary differential equations over changing constant values 但方法和结果不是我想要的 我没有解决方案,不知道如何包含这个周期性变化的值并解决系统。
或者这种情况下 odeint 不可用?
提前感谢任何答案。
【问题讨论】:
标签: python differential-equations