SageMath手动求解多元函数临界点时solve函数报TypeError错误

问题复现 #

编写教学用的多元函数临界点识别代码，手动实现「计算梯度→求解梯度零点」流程，初始代码如下：

var('x1 x2')
f(x1,x2) = x1^2 + x2^2
Df=f.derivative()
solve(Df(x1,x2)==(0,0),(x1,x2))

单独执行Df(x1,x2)可以正确返回梯度结果(2*x1, 2*x2)，但执行求解步骤时抛出如下错误：

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-33-bc76a33e3a15> in <module>
----> 1 solve(Df(x1,x2)==(0,0),(x1,x2))

/var/tmp/sage-jc4b6yulaujayb9sr94ia88eourzeqip0oidmas3/local/lib/python3.8/site-packages/sage/symbolic/relation.py in solve(f, *args, **kwds)
   1045 
   1046     if not isinstance(f, (list, tuple)):
-> 1047         raise TypeError("The first argument must be a symbolic expression or a list of symbolic expressions.")
   1048 
   1049     # f is a list of such expressions or equations

TypeError: The first argument must be a symbolic expression or a list of symbolic expressions.

报错原因 #

SageMath不会自动将元组之间的相等判断拆分为多个分量等式，Df(x1,x2)==(0,0)返回的是整体元组相等的判断对象，不是solve要求的「单个符号表达式」或「符号表达式/方程构成的列表/元组」，因此触发类型校验错误。

修复方案 #

将梯度的每个分量单独构造等于0的方程，组成列表传入solve即可，支持两种写法：

• 手动拆分分量（适合低维函数场景，逻辑直观适合教学演示）

var('x1 x2')
f(x1,x2) = x1^2 + x2^2
Df = f.derivative()
# 传入两个分量分别等于0的方程列表
solve([Df[0](x1,x2) == 0, Df[1](x1,x2) == 0], (x1,x2))

• 列表推导自动生成方程（适配任意元数的多元函数，代码更通用）

var('x1 x2')
f(x1,x2) = x1^2 + x2^2
Df = f.derivative()
# 自动遍历梯度所有分量生成对应方程
eq_list = [comp == 0 for comp in Df(x1,x2)]
solve(eq_list, (x1,x2))

两种写法运行后都会正确返回临界点结果：[[x1 == 0, x2 == 0]]。

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