有没有办法在 sympy plot3d plot 上放置彩条？

Question

所以我有一个 3d plot 我从 sympy 符号构建的，如文档中所示，让我们说代码是

from sympy import symbols
from sympy.plotting import plot3d
x, y = symbols('x y')
plot3d(x*y, (x, -5, 5), (y, -5, 5))

如何将颜色条附加到它？

Answer 1

使用这篇文章中的代码，您可以将 plot 移动到 matplotlib 并使用 matplotlib 绘制颜色条：

from sympy import symbols
from sympy.plotting import plot3d
from matplotlib import pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

x, y = symbols('x y')
plot1 = plot3d(x*y, (x, -5, 5), (y, -5, 5), show=False)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

backend = plot1.backend(plot1)
backend.ax = ax
backend._process_series(backend.parent._series, ax, backend.parent)
plt.close(backend.fig)
ax.collections[0].set_cmap('inferno') # optionally change the colormap
plt.colorbar(ax.collections[0])
plt.show()