[英]How to solve a differential equation with a non-constant coefficient using Python ?
假设有一个关于弹力绳长度的方程(用 x 表示),它取决于 object 的质量,例如播放器(用 m 表示)。
假设弹力绳的自然长度为30米,即起始position为x(0)=-30。
弹力绳的长度方程由下式给出:
x''(m) = g + b/m*x(m) -a1/m*x'(m) - a2*|x'(m)|*x'(m)
其中g, a1, a2
是常数; b
是一个步骤 function:当x<0
时b = -k
(另一个常数),当x>=0
0 时b = 0
0 。
import numpy as np
from scipy.integrate import odeint
g = 9.8
a1, a2 = 0.6, 0.8
k = 20
b = [-k, 0]
def dx_dt(x, m):
if x[0]>=0:
return [x[1], g+b[0]/m*x[0]-a1/m*x[1]-a2/m*np.abs(x[1])*x[1]]
else:
return [x[1], g+b[1]/m*x[0]-a1/m*x[1]-a2/m*np.abs(x[1])*x[1]]
init = [[-30, 0], [-40, 0.0001]]
m = np.linspace(1, 100, 10000)
fig, ax = plt.subplots(1, 2, sharey=True, figsize=(6, 4))
for i in range(len(init)):
xs = odeint(dx_dt, init[i], m)
ax[i].plot(m, xs[:, 0], 'r-')
ax[i].set_xlabel('mass (m)')
ax[i].set_ylabel('length (x)')
ax[i].set_xlim(xmin, xmax)
ax[i].set_ylim(ymin, ymax)
ax[i].set_title('init={}'.format(init[i]))
正确答案应该是正弦曲线
但是上面代码的结果是
代码有问题吗?
将长度坐标x
更改为向上,没有跳线的电线在 position 0 处的 rest 处,因此对于x<0
,电线的行为类似于 Z2A2D595E6ED9A0B24F027F2B63B134D6。 然后重力也指向下方。 修改后的 ODE function 是
def dx_dt(x, t, m):
acc = -g-a1/m*x[1]-a2/m*np.abs(x[1])*x[1]
if x[0] < 0: acc -= k/m*x[0]
return [x[1], acc]
绘制 3 个不同的质量
for i,ini in enumerate(init):
for m in masses:
xs = odeint(dx_dt, ini, t, args=(m,))
ax[i].plot(t, xs[:, 0], label="m=%.2f"%m)
然后给出图片
例如,如果第一种情况下的地面高度为-80m
,总高度为110m
,则跳线的质量必须小于 90kg。 对于更精确的陈述,使用内插和外推或数值求解器来找到x'(t)=0
的第一次时间和当时x(t)=ground
的临界质量。
显然,在任何情况下,跳线都不会重新进入“自由落体”阶段x>0
。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.