如何将数学表达式作为函数参数传入？

我希望将mathematical_operation作为函数的输入参数，让用户可以输入任意运算来获取结果。原本想用默认参数实现，写了如下代码：

mathematical_operation = input()
def fun(a=1, b=1, c=1, mathematical_operation=a*(x**2)+b*x+c):
    x = np.linspace(-1,1,10)
    y = (lambda a,b,c: mathematical_operation)(a,b,c)
    print(x)
    print(y)
    return y
fun(mathematical_operation = a*(x**2)+b*x+c)

但运行后出现报错：

Traceback (most recent call last):
  File "C:\Users\Documents\Python\test.py", line 38, in <module>
    def fun(a = 1, b = 1, c = 1, equation = a*(x**2)+b*x+c):
NameError: name 'a' is not defined

请问该如何解决此问题？

问题根源分析 #

你的代码报错的核心原因是Python函数的默认参数是在函数定义时求值的，而不是调用时。当你定义fun函数时，a、x这些变量都还没有被定义（x是函数内部才创建的，a作为参数此时还没绑定到具体值），所以Python会抛出NameError。

另外，你试图用lambda传递运算逻辑的思路也有问题——你的lambda并没有接收x作为参数，而且引用的mathematical_operation是定义时的外部变量，完全没起到动态传入运算的作用。

正确实现方案 #

这里提供两种安全且符合Python风格的实现方式，你可以根据需求选择：

方案1：传入Lambda函数（推荐，安全可控）

让用户传入一个lambda函数（或自定义函数），该函数接收a、b、c、x作为参数，返回运算结果。函数内部生成x后，调用传入的运算函数即可：

import numpy as np  # 别忘了导入numpy

def fun(a=1, b=1, c=1, mathematical_operation=None):
    # 设置默认运算：如果用户没传，就用二次函数
    if mathematical_operation is None:
        mathematical_operation = lambda a, b, c, x: a * (x**2) + b * x + c
    
    x = np.linspace(-1, 1, 10)
    y = mathematical_operation(a, b, c, x)
    
    print(x)
    print(y)
    return y

# 示例1：使用默认的二次函数
fun()

# 示例2：自定义运算（比如三次函数）
fun(mathematical_operation=lambda a,b,c,x: a * x**3 + b * x + c)

# 示例3：更复杂的自定义运算
fun(mathematical_operation=lambda a,b,c,x: np.sin(a*x) + b*np.cos(c*x))

方案2：传入字符串表达式（灵活但有安全风险）

如果你希望用户直接输入字符串形式的运算（比如输入"a*(x**2)+b*x+c"），可以使用eval来执行表达式，但注意：eval会执行任意代码，仅在你完全信任输入来源的情况下使用：

import numpy as np

def fun(a=1, b=1, c=1, mathematical_operation_str=None):
    x = np.linspace(-1, 1, 10)
    
    # 设置默认表达式
    if mathematical_operation_str is None:
        expr = "a*(x**2)+b*x+c"
    else:
        expr = mathematical_operation_str
    
    # 用eval计算，通过字典限定可用变量，降低风险
    y = eval(expr, {"__builtins__": None}, {"a":a, "b":b, "c":c, "x":x})
    
    print(x)
    print(y)
    return y

# 让用户输入运算表达式
user_expr = input("请输入数学运算（可用变量：a、b、c、x）：")
fun(mathematical_operation_str=user_expr)

关键总结 #

• 永远不要在函数默认参数中引用未定义的变量，默认参数的求值时机是函数定义时，不是调用时。
• 传递动态运算逻辑时，优先使用函数（lambda或自定义函数）的方式，安全且符合Python的函数式编程思想。
• 如果必须使用字符串表达式，一定要限制eval的作用域，避免安全漏洞。

内容的提问来源于stack exchange，提问作者Gerard