用sympy评估包含数值表达式的字符串?
我写了一个函数,用来计算拉格朗日插值多项式的数值表达式:
#!/usr/bin/env python
#coding: utf8
from sympy import *
import json
def polinomioLagrange(Xs, Ys, t):
x = Symbol('x')
expresion = ''
for k in range(len(Xs)):
if k >0: #Si no es el primero ni el último término de la sumatoria
expresion = expresion + '+' + str(Ys[k]) + '*'
elif k==0:
expresion = expresion + str(Ys[k]) + '*'
expresion = expresion + '('
for i in range(len(Xs)):
if k==i:
continue # Si i==k saltamos esta iteración para eliminar división sobre cero
expresion = expresion + '(' + '3' + '-' + str(Xs[i]) + ')'
if k != len(Xs)-1 and i!= len(Xs)-1:
expresion=expresion+'*'
#expresion = expresion + '(' + str(a) + '-' + str(Xs[i]) +' )' + '/' + '(' + str(Xs[k]) + '-' + str(Xs[i]) + ')'
expresion = expresion + '/'
for i in range(len(Xs)):
if k==i:
continue # Si i==k saltamos esta iteración para eliminar división sobre cero
expresion = expresion + '(' + str(Xs[k]) + '-' + str(Xs[i]) + ')'
if i != len(Xs)-1 and k != len(Xs)-1:
expresion=expresion+'*'
print expresion
print k, i
ewa = raw_input('Prompt :')
expresion = expresion + ')'
print expresion
当我用 lagrange([0,1,2,4],[-1,0,7,63],3)
调用这个函数时,得到了以下结果:
7*((3-1)*(3-2)*(3-4)/(0-1)*(0-2)*(0-4))+0*((3-0)*(3-2)*(3-4)/(1-0)*(1-2)*(1-4))+-1*((3-0)*(3-1)*(3-4)/(2-0)*(2-1)*(2-4))+63*((3-0)(3-1)(3-2)/(4-0)(4-1)(4-2))
这个结果其实是没问题的,不过,当我尝试用 sympify(expresion)
时,却出现了错误:
code, global_dict, local_dict) # take local objects in preference File "<string>", line 1, in <module> TypeError: 'NegativeOne' object is not callable
我明白可能是因为字符串里有一个 -1 的原因,但是……我该怎么简单地让这个表达式被计算出来呢?
1 个回答
1
这段话主要讲的是表达式(目前是string
)需要有正确的格式:
我的算法有问题
结果是这样的:
7*((3-1)*(3-2)*(3-4)/(0-1)*(0-2)*(0-4))+0*((3-0)*(3-2)*(3-4)/(1-0)*(1-2)*(1-4))+-1*((3-0)*(3-1)*(3-4)/(2-0)*(2-1)*(2-4))+63*((3-0)(3-1)(3-2)/(4-0)(4-1)(4-2))
最后一项缺少了一些*
:
如果我这样做:
from sympy import *
expression = '7*((3-1)*(3-2)*(3-4)/(0-1)*(0-2)*(0-4))+0*((3-0)*(3-2)*(3-4)/(1-0)*(1-2)*(1-4))+-1*((3-0)*(3-1)*(3-4)/(2-0)*(2-1)*(2-4))+63*((3-0)*(3-1)*(3-2)/(4-0)*(4-1)*(4-2))'
y = S(expression)
siympify(y)
那样就能正常工作,我也能得到我想要的结果。