如何在Python的欧拉方法函数中传入任意方程以提升复用性？

我想把欧拉方法写成递归函数，让它能接收任意由x和y组成的方程来提升复用性。比如当前代码里用的方程是x + y / 7，我希望能传入(x² + y) / 2这类任意形式的方程。想问能不能用numpy实现，或者有没有手写方案？我目前想到的思路是传入方程字符串，在函数内部转换重构来生成结果。

原代码：

y = -4
x = 0
step = 1

def eulersMethod(y, x, step):
    y = y+step*(x+y/7.)
    if y < 7:
        tmp = eulersMethod(y, x+step, step)
        if tmp < 7:
            y = tmp
    return y

print(eulersMethod(y, x, step))

方案1：直接传入函数（最推荐、安全） #

不用字符串传方程，直接把微分方程写成Python函数传进去，这是最简洁且无安全风险的方式，完全满足“任意方程”的需求。

修改后的递归欧拉函数：

def recursive_euler(y, x, step, dydx):
    # 计算当前步更新后的y值
    new_y = y + step * dydx(x, y)
    # 触发终止条件：y >=7时不再递归
    if new_y >= 7:
        return new_y
    # 递归调用，更新x和y的取值
    tmp = recursive_euler(new_y, x + step, step, dydx)
    return tmp if tmp < 7 else new_y

# 测试不同微分方程
if __name__ == "__main__":
    # 原方程：dy/dx = x + y/7
    def eq1(x, y):
        return x + y / 7.
    # 自定义新方程：dy/dx = (x² + y)/2
    def eq2(x, y):
        return (x**2 + y) / 2
    
    y0 = -4
    x0 = 0
    step = 1
    
    print("原方程结果：", recursive_euler(y0, x0, step, eq1))
    print("新方程结果：", recursive_euler(y0, x0, step, eq2))

方案2：字符串解析方程（需注意安全） #

如果一定要用字符串形式传入方程，可以用eval解析，但绝对不要传入不可信的字符串（避免代码注入风险）。

实现代码：

def recursive_euler_str(y, x, step, eq_str):
    # 用eval解析字符串方程，绑定x和y的当前值
    dydx_val = eval(eq_str, {"x": x, "y": y})
    new_y = y + step * dydx_val
    if new_y >= 7:
        return new_y
    tmp = recursive_euler_str(new_y, x + step, step, eq_str)
    return tmp if tmp < 7 else new_y

# 测试
if __name__ == "__main__":
    y0 = -4
    x0 = 0
    step = 1
    
    eq_str1 = "x + y/7."
    eq_str2 = "(x**2 + y)/2"
    
    print("原方程字符串版结果：", recursive_euler_str(y0, x0, step, eq_str1))
    print("新方程字符串版结果：", recursive_euler_str(y0, x0, step, eq_str2))

方案3：结合numpy向量化运算（适合批量计算） #

如果需要处理数组形式的x/y，或者想利用numpy的向量化能力提升效率，可以基于numpy实现方程逻辑：

import numpy as np

def recursive_euler_np(y, x, step, dydx):
    new_y = y + step * dydx(x, y)
    if new_y >= 7:
        return new_y
    tmp = recursive_euler_np(new_y, x + step, step, dydx)
    return tmp if tmp < 7 else new_y

# 用numpy实现的方程示例
def eq_np(x, y):
    return (np.square(x) + y) / 2

# 测试
if __name__ == "__main__":
    y0 = -4.0
    x0 = 0.0
    step = 1
    
    print("numpy版方程结果：", recursive_euler_np(y0, x0, step, eq_np))

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