odeint -(TypeError: can't convert expression to float ) - 将表达式导入 function 以执行 odeint

Question

这是我得到的错误

Traceback (most recent call last):

  File "C:\Users\user\.spyder-py3\Numerical Methods Problems\FreeFall.py", line 40, in <module>
    ans=odeint(vel,0,t)

  File "C:\ProgramData\Anaconda3\lib\site-packages\scipy\integrate\odepack.py", line 245, in odeint
    int(bool(tfirst)))

  File "C:\ProgramData\Anaconda3\lib\site-packages\sympy\core\expr.py", line 325, in __float__
    raise TypeError("can't convert expression to float")

TypeError: can't convert expression to float

这是我的代码-我对编码很陌生，并且正在学习将其用于数值计算：

from scipy.integrate import odeint
from sympy import *
import numpy as np
import math

def diff_cd(re_no):
    Re=Symbol('Re')
    expr=(24/Re)+(6/(1+(Re**0.5)))+0.4
    ans=diff(expr,Re).subs(Re,re_no)    
    return ans

def diff_re(k,u_no):
    u=Symbol('u')
    expr=k*u
    ans=diff(expr,u).subs(u,u_no)
    return ans    

ans = [diff_cd(20),diff_re(11,15)]

rhog=1.2
mug=1.872*(10**(-5))
a=0.3

u=Symbol('u')
pi=math.pi

k=(2*rhog*a/mug)
Re=k*u

p1=(rhog*pi*(a**2)*u*Re*((24/Re)+(6/(1+(Re**0.5)))+0.4))+(0.5*rhog*pi*(a**2)*(u**2)*diff_cd(Re)*diff_re(k,u))

ansfu=p1*(-1/24)


def vel(y,t):
    dudt = ansfu
    return dudt

t=np.linspace(1,100,100)
ans=odeint(vel,0,t)


print(ans)

我只需要在没有这个错误的情况下得到答案。 还有一种方法可以在单个 function 中完成所有这些操作

Answer 1

如果我在ode调用之前添加一个print(vel(0,0))我得到

-271.868595022194*u**2*(-3.97723522060237e-7*u**(-0.5)/(u**0.5 +
0.00509901951359279)**2 - 1.6224e-8/u**2) - 
543.737190044387*u**2*(0.4 + 6/(196.116135138184*u**0.5 + 1) + 0.000624/u)

也就是说，一个sympy的表达，而不是一个数字。 odeint无法使用它！

u被定义为Symbol ，因此任何使用它的 python 表达式也将是一个 sympy 表达式。

特别是如果您是编码新手，您应该坚持使用 package 或其他。 如果以符号方式定义 function，则使用sympy及其自己的求解器。 但是，如果scipy产生的数值解很重要，那么使用 python/numpy 定义 function。 不要试图混合 sympy 和 numpy （没有更多经验）。

Answer 2

老问题，但似乎你可以使用 sympy 的lambdify function 解决这个问题。 你会这样做：

f = lambdify(x, x**2, modules=['scipy'])

现在f是 lambda，您可以将其与 scipy 一起使用。