在 SymPy 中绘制具有复杂变量的隐式方程

Question

我正在尝试在 python 中绘制 SymPy 复数。

假设我想绘制复数单位圆，实部在 x 轴上，虚部在 y 轴上。查看我的代码：

from sympy import symbols, Eq
from sympy.plotting import plot_implicit

z = symbols('z')
expr = Eq(abs(z), 1)

p1 = plot_implicit(expr)

我通过这个方程定义单位圆：abs(z)=1。然后我使用 plot_implicit 绘制表达式。我得到两条垂直线穿过 z=-1 和 z=1，z 在 x 轴上，f(z) 在 y 轴上。

怎样才能画出满意的复单位圆？

Answer 1

如果 plot_implicit 可以直接使用一个复杂符号 z 就好了，但目前还不能：期望用户提供一个包含两个真实符号的表达式。这样做，通过引入真实的 x, y 并使 z 成为具有 x 和 y 的表达式。

from sympy import symbols, Eq, I
from sympy.plotting import plot_implicit

x, y = symbols('x y', real=True)
z = x + I*y
expr = Eq(abs(z), 1)    
p1 = plot_implicit(expr)

在 SymPy 中绘制具有复杂变量的隐式方程

Plot an implicit equation with a complex variable in SymPy

python

plot

sympy

complex-numbers