更改 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")

Answer 2

为了让它与最新的 sympy 版本（ 1.7.1 ）一起工作，我需要一些小的改变：

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[0].collections[0].set_cmap("magma")
p._backend.ax[0].figure.show()