在同一图表中绘制2个或更多功能

时间:2013-11-12 14:37:28

标签: python plot sympy

我想在同一图中用以下代码绘制2条曲线:

import sympy as syp


x, y = syp.symbols('x, y')

my_function = syp.exp(-(x-2)**2)*syp.exp(-(y-3)**2) + 2*syp.exp(-(x+1)**2)*syp.exp(-(y-1)**2) 

gradient_1 = syp.diff(my_function, x)
gradient_2 = syp.diff(my_function, y)


curve_1 = syp.plot_implicit(syp.Eq(gradient_1, 0))
curve_2 = syp.plot_implicit(syp.Eq(gradient_2, 0))

我所看到的只是第一个情节,而我想让两条曲线在同一张图片中,如果可能的话也可以使用网格。 有什么想法吗?

注意:使用matplotlib非常简单,但我找不到函数syp.plot_implicit

的任何具体示例

3 个答案:

答案 0 :(得分:6)

另一种可能更有效的方法是使用Or

同时计算两者
plot_implicit(Or(Eq(gradient_1, 0), Eq(gradient_2, 0)))

答案 1 :(得分:1)

如果您这样做可能会有效:

>>> curve_1.extend(curve_2)
>>> curve_1.show()

然而,混合隐式图可能尚未实现。

请注意,您的curve_1curve_2不是同情的“单曲线”即Series实例,而是“多条曲线的集合”,即{{1实例。

您还可以从Plot和其他curve_1._backend.fig属性中提取matplotlib对象。

总之,有一个很好的API可以做你想要的,但可能它背后的方法还没有完成。

答案 2 :(得分:0)

另一种方式:

curve_1.append(curve_2[0]) curve_1.show()