sympy求导/积分时如何将逗号分隔输入转为可变参数

错误原因 #

报错本质是两个核心问题：

• 直接传入未处理的逗号分隔参数字符串，SymPy无法识别多个独立的求导/积分变量
• 拆分后的字符串列表既不是SymPy可识别的Symbol符号对象，也没有作为可变参数传入函数

修复步骤 #

• 第一步：将用户输入的函数字符串用sympify()显式转换为SymPy可运算的表达式对象
• 第二步：对参数输入按逗号拆分，逐个处理每个元素：优先尝试转为整数（对应求导阶数），转换失败则转为Symbol符号对象
• 第三步：用Python可变参数解包语法*将处理好的参数列表传入diff/integrate函数

完整修改后的代码 #

from sympy import *

mode = input("Differentiate (1) or Integrate (2)?\t")
function = input("Function:\t")
parameter = input("With respect to:\t")

# 转换函数表达式为SymPy可识别格式
expr = sympify(function)

# 处理参数输入
param_list = []
for p in parameter.split(','):
    p = p.strip()
    # 优先尝试转整数，对应求导阶数场景
    try:
        param = int(p)
    except ValueError:
        # 转失败则动态创建符号对象
        param = symbols(p)
    param_list.append(param)

if mode == str(1):
    # 解包参数列表传入求导函数
    deriv = diff(expr, *param_list)
    print(deriv)
if mode == str(2):
    # 解包参数列表传入积分函数
    intg = integrate(expr, *param_list)
    print(intg)

效果验证 #

• 当输入参数为x, 2时，处理后参数列表为[x, 2]，解包后等价于diff(expr, x, 2)，完全兼容原有二阶求导用法
• 当输入参数为x, y, z时，处理后参数列表为[x, y, z]，解包后等价于diff(expr, x, y, z)，可正常完成多变量依次求导

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