python-3.x - 如何将 sympy 符号图与基于数据的散点图/线图结合起来？

Question

我正在测试优化算法，并希望查看决策变量在成本函数和约束图上的演变。例如下面的代码:

import matplotlib.pyplot as plt
from sympy import symbols, plot_implicit
from sympy.plotting.plot import Plot, ContourSeries
from sympy.utilities.lambdify import lambdify

(x1, x2) = symbols('x1 x2')

func = x1**4 + x2**4

p1 = Plot(ContourSeries(func,(x1,-1,5),(x2,-1,5)))
p1.extend(plot_implicit(x1 < 0,(x1,-1,5),(x2,-1,5),line_color='red',show=False))
p1.extend(plot_implicit(x2 < 2,(x1,-1,5),(x2,-1,5),line_color='red',show=False))

p1.show()

生成成本函数的等高线图，并将一些约束显示为红色区域，但我如何向该图中添加一个点？以下内容:

x1 = [4,4]
x2 = [4,4]
plt.scatter(x1,x2)

创建一个新的独立绘图。

Answer 1

这似乎并没有得到 sympy 的实际支持，但以下确实有效:

p1._backend.ax.scatter(x1, x2)
p1._backend.save('im.png')