python中画离散数据的频谱图
时间: 2023-07-24 22:11:21 浏览: 144
使用python实现离散时间傅里叶变换的方法
要画离散数据的频谱图,需要使用numpy和matplotlib库。
假设你有一组实部为`real_data`、虚部为`imag_data`的复数数据。首先,使用numpy将实部和虚部合并成复数数组:
```python
import numpy as np
# 将实部和虚部合并成复数数组
complex_data = np.array(real_data) + 1j * np.array(imag_data)
```
接着,使用numpy计算数据的离散傅里叶变换(DFT):
```python
# 计算数据的DFT
spectrum = np.fft.fft(complex_data)
```
然后,使用numpy计算频率:
```python
# 计算频率
freq = np.fft.fftfreq(len(spectrum))
```
最后,使用matplotlib画出频谱图:
```python
import matplotlib.pyplot as plt
# 画出频谱图
plt.stem(freq, np.abs(spectrum))
plt.title('Spectrum plot of discrete data')
plt.xlabel('Frequency')
plt.ylabel('Magnitude')
plt.show()
```
这样就可以画出离散数据的频谱图了。
阅读全文