在Matlab的Ezplot中绘制python/pyplot中的任意二维函数

我正在寻找一种方法来生成类似于ezplot如何在MATLAB中工作的绘图,因为我可以输入:

ezplot('x^2 + y^2 = y + 5')

并获得准备好任意函数的图形.我只担心我有斧头和斧头的情况.

我只有这个功能,而且我真的不想尝试计算某些给定x范围的所有y值,如果我不需要的话.

我见过的几个解决方案是关于决策边界(这不是.没有测试数据或任何东西,只是一个任意函数)或者都是已经定义为y =某些x方程的函数帮我.

我有点接受,如果有一个很好的方法来模仿Wolfram | Alpha的求解功能("求解x ^ 2 + y ^ 2 = y + 5为y"将给我两个函数,然后我可以单独绘制),而是更喜欢ezplot,因为它在MATLAB中或多或少是即时的.

我认为您可以使用sympyplotting和parse_expr为此例如,这将如下工作

from sympy.plotting import plot_implicit
from sympy.parsing.sympy_parser import parse_expr

def ezplot(s):
    #Parse doesn't parse = sign so split
    lhs, rhs = s.replace("^","**").split("=")
    eqn_lhs = parse_expr(lhs)
    eqn_rhs = parse_expr(rhs)

    plot_implicit(eqn_lhs-eqn_rhs)

ezplot('x^2 + y^2 = y + 5')

这可以根据需要进行