如何序列化含菲涅尔积分的mpmath lambdify SymPy函数？

问题核心 #

你需要生成能处理负输入（涉及平方根但输出为实数）的菲涅尔积分函数，同时要能序列化保存到磁盘：

• 用["scipy", "numpy"]后端lambdify生成的函数可正常序列化，但负输入会产生复数，不符合需求；
• 用mpmath后端生成的函数能正确处理负输入，但序列化（dill/cloudpickle）失败，简单函数加recurse=True可行，但菲涅尔积分函数会触发递归深度超限或加载报错。

以下是几种可行的解决方案：

方案一：用numpy/scipy后端，封装处理复数转实数 #

既然numpy/scipy生成的函数可正常序列化，且你确认输出始终为实数，只需在函数外层封装一层，将复数结果转为实数即可：

import sympy as sm
import dill
import numpy as np

a, b = sm.symbols('a b')
f_fresnel = sm.integrate(sm.cos(a*b**2), (b, 0, 1))
# 用默认后端生成函数
F_fresnel_np = sm.lambdify((a, b), f_fresnel, ["scipy", "numpy"])

# 封装函数，将复数结果转为实数
def wrapped_fresnel(a, b):
    result = F_fresnel_np(a, b)
    return np.real(result) if isinstance(result, complex) else result

# 序列化封装后的函数
with open('fresnel_func.pkl', 'wb') as f:
    dill.dump(wrapped_fresnel, f)

# 加载测试
with open('fresnel_func.pkl', 'rb') as f:
    loaded_func = dill.load(f)

print(loaded_func(-1, 0))  # 正常输出实数结果

方案二：自定义mpmath函数，避开内部上下文绑定 #

直接提取lambdify生成函数的核心逻辑，封装成纯Python函数，避免绑定mpmath的内部上下文对象：

import sympy as sm
import dill
import mpmath as mp

a, b = sm.symbols('a b')
f_fresnel = sm.integrate(sm.cos(a*b**2), (b, 0, 1))
# 先查看lambdify生成的函数代码，提取核心逻辑
F_fresnel = sm.lambdify((a, b), f_fresnel, "mpmath")

# 手动封装纯函数，直接调用mpmath的计算逻辑
def custom_fresnel(a, b):
    # 复制lambdify生成的菲涅尔积分计算逻辑
    return mp.quad(lambda t: mp.cos(a * t**2), [0, 1])

# 序列化自定义函数
dill.dump(custom_fresnel, open('custom_fresnel.pkl', 'wb'))
loaded_func = dill.load(open('custom_fresnel.pkl', 'rb'))

print(loaded_func(-2, 1))  # 正常计算负输入

方案三：序列化SymPy表达式，加载后重新生成函数 #

不序列化lambdify后的函数，而是序列化SymPy的原始表达式，加载后再重新执行lambdify，完全避开函数序列化的问题：

import sympy as sm
import dill

a, b = sm.symbols('a b')
f_fresnel = sm.integrate(sm.cos(a*b**2), (b, 0, 1))

# 序列化SymPy表达式
dill.dump(f_fresnel, open('fresnel_expr.pkl', 'wb'))

# 加载后重新生成函数
loaded_expr = dill.load(open('fresnel_expr.pkl', 'rb'))
F_fresnel = sm.lambdify((a, b), loaded_expr, "mpmath")

print(F_fresnel(-1, 0))  # 正常计算

问题根源说明 #

mpmath后端生成的函数会绑定mpmath内部的上下文对象（如ivmpf），这些对象结构复杂且存在循环引用，导致dill递归序列化时触发深度超限；cloudpickle虽能保存，但加载时无法正确重建mpmath的内部数值类型（如mpq），因此报错。

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