用Python中的Sympy评价符号微分

时间:2014-02-08 12:35:59

标签: python sympy

我是python中的codig NewtonRaphson算法,使用Sympy库,这是我的算法实现:

def NewtonRaphson(fx,p0,tolerancia,iteracionesMaximas):
    print
    fx = S(fx)
    x = Symbol('x')
    i = 1
    p = 0.0

    while i<= iteracionesMaximas:

        y = fx.subs(x,p0)
        yy = diff(fx,x).subs(x,p0)
        p = p0 - (y/yy)

        if absolute(p-p0) < tolerancia:
            print "Se encontró la raíz y={raiz} luego de {n} iteraciones".format(raiz=p, n=i-1)
            return

        i += 1
        print "{i}\t{p}\t{p0}".format(i=i-1,p=p,p0=p0)
        p = p0

    print "El método falló luego de {i} iteraciones".format(i=i-1)

我正在接受以下消息的加薪:

line 18, in NewtonRaphson    
    yy = diff(fx,x).subs(x,p0)    
  File "/usr/local/lib/python2.7/dist-packages/numpy/lib/function_base.py", line 984, in diff    
    if n < 0:    
  File "/usr/local/lib/python2.7/dist-packages/sympy/core/relational.py", line 226, in __nonzero__    
    raise TypeError("symbolic boolean expression has no truth value.")    
TypeError: symbolic boolean expression has no truth value.    

我将fx函数称为等于'x**3-x-1'的字符串,并输入调用NewtonRaphson(fx,1.7,10**(-4),17)

我有什么问题?

1 个答案:

答案 0 :(得分:1)

看起来你正在将符号表达式传递给numpy的绝对函数。 根据您的代码判断,我敢打赌您的导入语句是

from pylab import *
from sympy import *

如果是这种情况,你应该用absolute替换Abs,这样你就可以使用内置的绝对值函数来代替numpy数组。

此外,您需要将p = p0替换为p0 = p才能使算法正常运行。

这是一个工作版本,其中import语句有所改变。 我还删除了一些不必要的陈述。

import sympy as sy

x = sy.symbols('x')
fx = x**3 - x - 1

def NewtonRaphson(fx,p0,tolerancia,iteracionesMaximas):

    for i in xrange(iteracionesMaximas):

        y = fx.subs(x, p0)
        yy = sy.diff(fx,x).subs(x, p0)
        p = p0 - (y / yy)

        if sy.Abs(p-p0) < tolerancia:
            print "Se encontró la raíz y={raiz} luego de {n} iteraciones".format(raiz=p, n=i-1)
            return

        print "{i}\t{p}\t{p0}".format(i=i+1,p=p,p0=p0)
        p0 = p

    print "El método falló luego de {i} iteraciones".format(i=i-1)

NewtonRaphson(fx,1.7,10**(-4),17)