【发布时间】:2011-03-17 17:29:38
【问题描述】:
我正在使用 python,我想要一个函数,该函数接受一个包含一个变量 (x) 的数学表达式的字符串,并返回一个使用 lambda 评估该表达式的函数。语法应该是这样的:
f = f_of_x("sin(pi*x)/(1+x**2)")
print f(0.5)
0.8
语法应该允许 ( ) 以及 [ ] 并使用标准运算符优先级。 Trig 函数的优先级应低于乘法,高于加法。因此字符串 'sin 2x + 1' 将等价于 sin(2x)+1,尽管两者都是有效的。这是用于评估代数和三角表达式的用户输入,因此请考虑数学语法而不是编程语法。支持的函数列表应该易于扩展,代码应该清晰易懂。不折叠常量表达式也没关系。
这里的示例函数不完整。它采用表示表达式的嵌套列表并生成适当的函数。虽然有点容易理解,但对于 python 来说,即使这样也很难看。
import math
def f_of_x(op):
if (isinstance((op[0]), (int, long, float, complex)) ):
return (lambda x:op[0])
elif op[0]=="pi": return lambda x: 3.14159265358979
elif op[0]=="e": return lambda x: 2.718281828459
elif op[0]=="x": return lambda x: x
elif op[0]=="sin": return lambda x: math.sin(f_of_x(op[1])(x))
elif op[0]=="cos": return lambda x: math.cos(f_of_x(op[1])(x))
elif op[0]=="tan": return lambda x: math.tan(f_of_x(op[1])(x))
elif op[0]=="sqrt": return lambda x: math.sqrt(f_of_x(op[1])(x))
elif op[0]=="+": return lambda x: (f_of_x(op[1])(x))+(f_of_x(op[2])(x))
elif op[0]=="-": return lambda x: (f_of_x(op[1])(x))-(f_of_x(op[2])(x))
elif op[0]=="*": return lambda x: (f_of_x(op[1])(x))*(f_of_x(op[2])(x))
elif op[0]=="/": return lambda x: (f_of_x(op[1])(x))/(f_of_x(op[2])(x))
elif op[0]=="**": return lambda x: (f_of_x(op[1])(x))**(f_of_x(op[2])(x))
# should never get here with well formed input
return
def test():
# test function f(x) = sin(pi*x)/(1+x**2)
s = ['/',['sin',['*',['pi'],['x']]],['+',[1],['**',['x'],[2]]]]
f = f_of_x(s)
for x in range(30):
print " "*int(f(x*0.2)*30+10)+"x"
作为一般准则,将您的解决方案视为有关 lambda 和解析器的教程 - 而不是代码高尔夫。示例代码就是这样,你觉得最清楚的就写吧。
【问题讨论】:
-
我不确定你的问题是什么。你似乎已经设定了一个挑战,然后迎接了它。您唯一没有的部分是可以从字符串生成列表表示的解析器。
-
你知道
sin(pi*x)/(1+x**2)在 x=0.5 实际上是 0.8,对吧? -
@Ned - 是的,这是一个挑战。我正在制作 lambda 的重要用途(动态创建任意复杂函数)的示例,并意识到我从未编写过完成它所需的那种解析器。我认为这对这里的某个人来说可能很有趣和/或容易。如果没有代码出现,我可能会追求答案中给出的选项来制作我自己的。 @Gabe - 修复它。
标签: python math parsing lambda