4

我正在尝试使用 Python2.7 数值求解方程。这是整个代码：

from sympy import *
from sympy import Symbol
from sympy.solvers import nsolve
from scipy import *
from pylab import *
import numpy as np

# Symbols
theta = Symbol('theta')
phi = Symbol('phi')
phi0 = Symbol('phi0')
H0 = Symbol('H0')
# Constants
a = 0.05 
b = 0.05**2/(8*pi*1e-7)
c= 0.001/(4*pi*1e-7)
phi0 = 60*pi/180 
H0 = -0.03/(4*pi*1e-7)
def m(theta,phi):
    return np.array([sin(theta)*cos(phi), sin(theta)*cos(phi), cos(phi)])
def h(phi0):
    return np.array([cos(phi0), sin(phi0), 0])
def k(theta,phi,phi0):
    return np.vdot(m(theta,phi),h(phi0))
def F(theta,phi,phi0,H0): 
    return -(a*H0)*k(theta,phi,phi0)+b*(cos(theta)**2)+c*(sin(2*theta)**2)+sin(theta)**4*sin(2*phi)**2
def F_phi(theta,phi,phi0,H0):
    return simplify(diff(F(theta,phi,phi0,H0),phi))
def G(phi):
    return F_phi(pi/2,phi,phi0,H0)
solution = nsolve(G(phi), phi)
print solution

这是我的回溯：

Traceback (most recent call last):
File "Test.py", line 33, in <module>
solution = nsolve(G(phi), phi)
File "Test.py", line 32, in G
return F_phi(pi/2,phi,phi0,H0)
File "Test.py", line 30, in F_phi
return simplify(diff(F(theta,phi,phi0,H0),phi))
File "Test.py", line 28, in F
return -(a*H0)*k(theta,phi,phi0)+b*(cos(theta)**2)+c*(sin(2*theta)**2)+sin(theta)**4*sin(2*phi)**2
File "Test.py", line 26, in k
return np.vdot(m(theta,phi),h(phi0))
File "Test.py", line 22, in m
return np.array([sin(theta)*cos(phi), sin(theta)*cos(phi), cos(phi)])
AttributeError: cos

我正在使用 sympy、numpy 和 pylab 库。所以，我没有遇到 cos 函数的问题。有什么帮助吗？

4

2 回答 2

9

问题是使用星形导入而不是在不同的命名空间下导入每个包。

sympy.functions.elementary.trigonometric.cos这在名称下导入函数cos：

from sympy import *

之后，<ufunc 'cos'>在 name 下导入cos，覆盖之前的定义：

from scipy import *

cos然后，它用完全相同的函数的另一个副本（来自 matplotlib 包）覆盖前一个函数：

from pylab import *

这也导入了相同<ufunc 'cos'>的名称，但np.cos名称相同。这是导入东西的正确方法：

import numpy as np

最后，你会得到一份 cos 函数的副本，它知道如何将自身应用于浮点数，而不是 sympy 对象。当您尝试将该功能应用于 sympy 对象时，phi您会得到AttributeError. 总而言之，这个特定问题的解决方案是修复导入并知道您是想要来自 sympy 的函数还是来自 numpy 的函数。

于 2012-11-12T00:08:46.623 回答
2

你导入cos函数了吗？它在math模块中

from math import cos

同样的事情sin。

于 2012-11-11T23:48:05.113 回答