我有一个8个微分方程组,有8个边界条件,我试图用scipy的solve_bvp来求解。下面你可以找到我正在使用的代码
import numpy as np
from scipy.integrate import solve_bvp
def fun(x, y):
x_dot = y[2]
y_dot = y[1]
vx_dot = 53441.25/(4106.89-15.23*x)*(-y[6]/np.sqrt(y[6]**2+y[7]**2))
vy_dot = 53441.25/(4106.89-15.23*x)*(-y[7]/np.sqrt(y[6]**2+y[7]**2))-3.986e14/(y[1]+6371.0e3)**2
lambda1_dot = 0
lambda2_dot = 0
lambda3_dot = -y[4]
lambda4_dot = -y[5]
return np.vstack((x_dot, y_dot, vx_dot, vy_dot, lambda1_dot, lambda2_dot, lambda3_dot, lambda4_dot))
def bc(ya, yb):
return np.array([ya[0]-113021.9, yb[1]-500e3, yb[2]-7616.5, yb[3], yb[4], ya[5]-lambda2_0, ya[6]-lambda3_0, ya[7]-lambda4_0])
lambda1_0 = 0.0
lambda2_0 = 0.1
lambda3_0 = 0.1
lambda4_0 = 0.1
tf = 300
x = np.linspace(0, tf, tf*2)
y = np.zeros((8, x.size))
y[0] = 113021.9
y[1] = np.linspace(127734.0, 500.0e3, tf*2)
y[2] = np.linspace(1854.2, 7616.5, tf*2)
y[3] = np.linspace(1479.1, 0, tf*2)
y[4] = lambda1_0
y[5] = lambda2_0
y[6] = lambda3_0
y[7] = lambda4_0
res_a = solve_bvp(fun, bc, x, y)
但是它看起来有问题,因为我收到了以下错误消息:ValueError:除了连接轴之外,所有输入数组的维度都必须完全匹配
如何解决这个问题?在
你需要解释一下
通过函数
^{pr2}$fun
运行完整数组y
,因此每个状态变量都包含一个用于同时进行多点计算的np.array
。返回值必须与此任务兼容,除了初始化y_dot = np.zeros_like(y)
之外,最简单的方法是通过相关问题 更多 >
编程相关推荐