python - 在复平面上绘制图形

Question

我想绘制函数的图形:w(s) = 1/(1+s)。

s 是虚数单位 (1j) 和名为 omega 的变量的乘积；即s = 1j*omega。

如何使用 Python(2.7 或 3.4)和 matplotlib 在复轴(实轴和虚轴)上绘制它？

Answer 1

您在问题中不清楚要绘制什么，但假设 omega = x + iy 是您希望绘制 w(s) 作为函数的数字，您必须决定如何呈现复数数字 w(s)。您可以选择用笛卡尔轴表示实轴 (x) 和虚轴 (y) 的绘图，并将 w(s) 的绝对值绘制为颜色，或者您可以选择分别绘制实部和虚部。 例如，

import matplotlib as plt
import numpy as np

x = np.linspace(-0.5,0.5,100)
y = np.linspace(-3,0,100)
X, Y = np.meshgrid(x,y)

def f(x, y):
    return 1./(1+1j*(x+1j*y))

import pylab
pylab.imshow(np.abs(f(X,Y)))
pylab.show()