python - 在 Matplotlib 的 stem 命令中使用复数

Question

我有两个 numpy 数组a（具有整数值）和b（具有复数）。现在当我使用时stem(a,b)，我收到以下错误：

C:\Python27\lib\site-packages\numpy\core\numeric.py:235: 
    ComplexWarning: Casting complex values to real discards the imaginary 
    part return array(a, dtype, copy=False, order=order) 
    Out[5]: <Container object of 3 artists>

有人可以帮忙吗？

Answer 1

你想让它做什么？ stemplot 在a 从基线到高度的每个水平位置绘制垂直线b。但这里，b是一个复数——你需要它是一个实值量。也许你想要绝对值，np.abs(b)？还是真实的部分，np.real(b)？也许有两个茎图，stem(a, np.real(b)); stem(a, np.imag(b))？

Answer 2

对于茎图，您只需要复数的实部。

x = np.arange(-64,64,1)
C = [cmath.exp(abs(i) * cmath.log(a3)) for i in x]
stem(x,real(C))

茎图