02 如何使用python做傅里叶变换（使用numpy库）

import numpy as np #载入numpy库
import matplotlib.pyplot as plt #载入matplotlib库的pyplot模块
#定义时域信号
fs=128 #采样率
t=np.arange(0,10,1/fs) #定义时间点
xt=5*np.sin(2*np.pi*10*t) #生成10Hz谐波，幅值为5

xf=np.fft.fft(xt) #傅里叶变换
af=np.abs(xf)/len(xf) #求复数的模
afn=np.short(af.size/2) #整型
f=np.arange(0,fs/2,fs/len(t)) #定义频率点
plt.subplot(211) #子图1
plt.plot(t,xt) #时域信号
plt.subplot(212) #子图2
plt.plot(f,2*af[:afn]) #频域信号