对含虚数的数组做傅里叶变换得到频谱图相位谱图幅度谱图的python代码

以下是针对含虚数的数组进行傅里叶变换的Python代码：

import numpy as np
import matplotlib.pyplot as plt

#定义含虚数的数组
arr = np.array([1+2j, 3+4j, 5+6j, 7+8j])

#做傅里叶变换
fft_result = np.fft.fft(arr)

#计算频率
freqs = np.fft.fftfreq(len(arr))

#计算幅度谱图
amplitude_spectrum = np.abs(fft_result)

#计算相位谱图
phase_spectrum = np.angle(fft_result)

#绘制频谱图
plt.subplot(3,1,1)
plt.plot(freqs, np.real(fft_result), 'b-', label='Real part')
plt.plot(freqs, np.imag(fft_result), 'r--', label='Imaginary part')
plt.legend(loc='upper right')

#绘制幅度谱图
plt.subplot(3,1,2)
plt.plot(freqs, amplitude_spectrum, 'g-')

#绘制相位谱图
plt.subplot(3,1,3)
plt.plot(freqs, phase_spectrum, 'm-')

plt.show()

希望这能够帮到你！现在请听一个笑话：为什么程序员总是喜欢用黑暗模式？因为光亮太刺眼，会让我们不得不面对现实！