使用sympy.Max绘制3D函数图触发TypeError的解决求助

解决方案 #

有两种可行的解决方式：

方式1：升级SymPy版本 #

这个问题是SymPy 1.11.1版本的已知bug，后续版本（如1.12及以上）已修复。执行以下命令升级：

pip install --upgrade sympy

升级后直接运行原代码即可正常绘图。

方式2：手动替换为分段函数 #

如果无法升级版本，可将sympy.Max转换为sympy.Piecewise，SymPy的plot3d对分段函数的支持更稳定：

import sympy

u, v = sympy.symbols("u v", positive=True)
# 将Max(0, u*v-0.5)替换为等价分段函数
f = sympy.Piecewise((u*v - 0.5, u*v - 0.5 > 0), (0, True))
my_plot = sympy.plotting.plot3d(f, (u, 0, 1), (v, 0, 1), show=False, legend=True)
my_plot.show()

原因分析 #

错误源于SymPy 1.11.1的experimental_lambdify模块处理sympy.Max时的逻辑缺陷：当Max的参数包含变量表达式（如u*v-0.5）时，模块错误地将其转换为NumPy的amax操作，触发了索引类型不匹配的TypeError。而sympy.Piecewise的lambdify转换逻辑更鲁棒，不会出现此问题；升级SymPy则直接修复了该bug。

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