Python 属性错误：cos

Question

我正在尝试使用 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 函数的问题。有什么帮助吗？

Answer 1

您是否导入了cos 函数？它在math 模块中

from math import cos

sin 也是如此。

Answer 2

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

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

from sympy import *

之后，你在cos这个名字下导入<ufunc 'cos'>，覆盖之前的定义：

from scipy import *

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

from pylab import *

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

import numpy as np

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