python求频谱代码

以下是Python中求频谱的代码示例：

import numpy as np
import matplotlib.pyplot as plt

# 生成信号
fs = 1000  # 采样率
t = np.arange(0, 1, 1/fs)  # 时间序列
f1 = 50  # 信号频率
f2 = 120
x = np.sin(2*np.pi*f1*t) + np.sin(2*np.pi*f2*t)

# 计算频谱
N = len(x)
X = np.fft.fft(x)
X_mag = np.abs(X)/N  # 频谱幅值
f = np.arange(0, fs/2, fs/N)  # 频率序列

# 绘制频谱图
plt.plot(f, X_mag[:N//2])
plt.xlabel('Frequency [Hz]')
plt.ylabel('Magnitude')
plt.show()