python - 在 SymPy 的 `` plot3d`` 中更改颜色图

Question

在 SymPy 中，您可以根据表达式自动绘制曲面，即

from sympy import symbols
from sympy.plotting import plot3d

x, y = symbols('x y')
monkey_saddle = x**3 - 3*x*y**2
plot3d(monkey_saddle, cmap="RdYlBu")

得到

我找不到任何改变颜色图的kwarg。我知道我可以创建一个 lambdify，然后直接使用 matplotlib 绘制绘图。但我很好奇是否有一种方法可以直接在 SymPy 中完成。

Answer 1

看了sympy.plotting.plot.py的源码，好像cmap设置成了jet:

collection = self.ax.plot_surface(x, y, z, 
                                  cmap=self.cm.jet, 
                                  rstride=1, cstride=1, 
                                  linewidth=0.1)

需要设置collections对象的cmap，调用plot3d()前调用unset_show()禁止调用pyplot.show():

from sympy import symbols
from sympy.plotting import plot3d
from sympy.plotting.plot import unset_show

unset_show()

x, y = symbols('x y')
monkey_saddle = x**3 - 3*x*y**2
p = plot3d(monkey_saddle)

p._backend.ax.collections[0].set_cmap("RdYlBu_r")