Python中如何输出含未知参数函数的符号表达式结果？

你当前使用的numpy、scipy属于数值计算类库，仅支持传入具体数值完成计算，要实现保留未知参数的符号运算，替换为SymPy符号计算库的对应接口即可。

完整实现代码 #

# 导入SymPy的符号定义、指数函数、朗伯W函数接口
from sympy import symbols, exp, lambertw

a = 1
b = 2
c = 3 

# 定义c_s为符号变量
c_s = symbols('c_s')

def my_func(c_s):
   # 替换原numpy的数值exp为SymPy的符号exp
   return a*b*exp(c_s*c**2)

# 输出符号形式的函数结果
print(my_func(c_s))
# 输出传入lambertw后的符号表达式
print(lambertw(my_func(c_s)))

运行效果 #

• 执行print(my_func(c_s))，输出结果为：2*exp(9*c_s)，完全符合需求
• 执行print(lambertw(my_func(c_s)))，输出结果为：LambertW(2*exp(9*c_s))，所有常量已自动计算完成，仅保留未知参数c_s

补充：兼容数值计算需求 #

如果后续需要将符号表达式转换为可传入数值计算的函数，可以使用SymPy的lambdify方法兼容numpy数值计算逻辑：

from sympy import lambdify
# 将符号表达式转为numpy兼容的数值计算函数
num_func = lambdify(c_s, my_func(c_s), 'numpy')
# 传入具体数值即可得到数值结果
print(num_func(0.1))

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