python - 用户输入一个函数

Question

我正在编写一个程序，涉及计算用户定义函数的最大值，但是如果用户根据变量(例如 x)输入函数，即

input = x**2 + 2*x + 1

我(完全符合预期)收到错误，因为 x 未定义。

有谁知道允许用户输入单个变量(例如 x)的函数的有效方法吗？

Answer 1

如果您不太担心安全性，最简单的解决方案是评估表达式，例如

def calculate(value, function):
    x = value
    return eval(function)

print calculate(2, "x**2 + 2*x + 1")
print calculate(2, "x**3 - x**2 + 1")

输出:

9
5

您可以通过传递一个空的builtin字典来使其更安全，但我仍然认为一些聪明的代码将能够访问代码的内部结构。