python - 在 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)