Question

我一直在使用odeint和边界条件。我想要做的就是解决这个图中给出的微分方程1

其中在我的代码中R = R，ph = Phi，al =α，a = a，m = m，l = 1且om = omega。我试图实现的初始条件是R（0）= O（r ^ l）;如果l / = 0，则Phi（0）= O（r ^ {l-1}），如果l = 0，则Phi（0）= O（r）; a（0）= 1且a（inf）= 1 / alpha（inf）（另外我需要R（inf）= 0）。我尝试应用拍摄方法，以便找到与我的边界条件最匹配的alpha的初始条件。我还需要找到最适合无限远R的边界条件的欧米茄。我写的代码如下：

import numpy as np
import matplotlib.pyplot as plt
from scipy.integrate import odeint
import time

start = time.clock()
def system_DE(IC,p,r):
    l = p[0]
    m = p[1]
    om = p[2]

    R = IC[0]
    ph = IC[1]
    a = IC[2]
    al = IC[3]

    dR_dr = ph
    da_dr = a*((2*l+1)*r/2*(om**2*a**2*R**2/al**2+ph**2+l*(l+1)*a**2*R**2/r**2+m**2*a**2*R**2)-(a**2-1)/(2*r))
    dal_dr = al*(da_dr/a-l*(l+1)*(2*l+1)*a**2*R**2/r-(2*l+1)*m**2*a**2*r*R**2+(a**2-1)/r)
    dph_dr = -2*ph/r-dal_dr*ph/al+da_dr*ph/a-om**2*a**2*R/al**2+l*(l+1)*a**2*R/r**2+m**2*a**2*R

    return [dR_dr,da_dr,dal_dr,dph_dr]

def init(u,p,r):
    if p==0:
        return np.array([1,r,1,u])
    else:
        return np.array([r**l,l*r**(l-1),1,u])

l = 0
m = 1
ep = 0.3
n_om = 10
omega = np.linspace(m-ep,m+ep,n_om)
r = np.linspace(0.0001, 100, 1000)


niter = 100
u = 0
tol = 0.1
ustep = 0.01

p = np.zeros(3)
p[0] = l
p[1] = m
for j in range(len(omega)):
    p[2] = omega[j]
    for i in range(niter):
        u += ustep
        Y = odeint(system_DE(init(u,p[0],r[0]),p,r), init(u,p[0],r[0]), r)
        print Y[-1,2]
        print Y[-1,3]
        if abs(Y[len(Y)-1,2]-1/Y[len(Y)-1,3]) < tol:
            print(i,'times iterations')
            print("a'(inf)) = ", Y[len(Y)-1,2])
            print('y"(0) =',u)
            break
    if abs(Y[len(Y)-1,0]) < tol:
        print(j,'times iterations in omega')
        print("R'(inf)) = ", Y[len(Y)-1,0])
        break

但是，当我运行它时，我得到：错误：函数及其雅可比行列必须是可调用的函数。

有人可以帮我理解我的错误是什么吗？

问候，

Luis Padilla。

Answer 1

首先，odeint的第一个参数是你的派生函数system_DE。只是传递它的名字，没有括号或参数。 Odeint在内部调用它并提供参数。

Answer 2

我修复了我的代码，现在它给了我一些结果。然而，当我运行它时，我得到一些警告，我不知道如何解决它。有人可以帮我解决吗？基本上我的代码是这样的：

$ git add *somefiles*
$ git commit
$ git checkout -b fix/new_branch
$ git push origin fix/new_branch

但是当我运行它时，我得到了这个：

import numpy as np
import matplotlib.pyplot as plt
from scipy.integrate import odeint
import time

def system_DE(IC,r,l,m,om):    
    R = IC[0]
    ph = IC[1]
    a = IC[2]
    al = IC[3]

    dR_dr = ph
    da_dr = a*((2*l+1)*r/2*(om**2*a**2*R**2/al**2+ph**2+l*(l+1)*a**2*R**2/r**2+m**2*a**2*R**2)-(a**2-1)/(2*r))
    dal_dr = al*(da_dr/a-l*(l+1)*(2*l+1)*a**2*R**2/r-(2*l+1)*m**2*a**2*r*R**2+(a**2-1)/r)
    dph_dr = -2*ph/r-dal_dr*ph/al+da_dr*ph/a-om**2*a**2*R/al**2+l*(l+1)*a**2*R/r**2+m**2*a**2*R

    return [dR_dr,dph_dr,da_dr,dal_dr]

def init(u,p,r):
    if p==0:
        return np.array([1.,r,1.,u])
    else:
        return np.array([r**p,l*r**(p-1),1,u])


l = 0.
m = 1.
ep = 0.2
n_om = 30
omega = np.linspace(m-ep,m+ep,n_om)
r = np.linspace(0.001, 100, 1000)


niter = 1000
tol = 0.01
ustep = 0.01

for j in range(len(omega)):
    print('trying with $omega =$',omega[j])
    p = (l,m,omega[j])
    u = 0.001
    for i in range(niter):
        u += ustep
        ini = init(u,p[0],r[0])
        Y = odeint(system_DE, ini,r,p,mxstep=500000)
        if abs(Y[len(Y)-1,2]-1/Y[len(Y)-1,3]) < tol:
            break
    if abs(Y[len(Y)-1,0]) < tol and abs(Y[len(Y)-1,2]-1/Y[len(Y)-1,3]) < tol:
        print(j,'times iterations in omega')
        print(i,'times iterations')
        print("R'(inf)) = ", Y[len(Y)-1,0])        
        print("alpha(0)) = ", Y[0,3])
        print("\omega",omega[j])
        break

plt.subplot(2,1,1)
plt.plot(r,Y[:,0],'r',label = '$R$')
plt.plot(r,Y[:,1],'b',label = '$d R /dr$')
plt.xlim([0,10])
plt.legend()
plt.subplot(2,1,2)
plt.plot(r,Y[:,2],'r',label = 'a')
plt.plot(r,Y[:,3],'b', label = '$alpha$')
plt.xlim([0,10])
plt.legend()
plt.show()

我该如何解决这个问题？

此致

Luis Padilla。

Python中的Odeint，射击方法和边界条件

2 个答案: