在Python中,使用numpy库进行傅里叶变换非常简单
import numpy as np
# 创建一个简单的信号
time = np.linspace(0, 1, 1000)
signal = np.sin(2 * np.pi * 5 * time)
# 对信号进行傅里叶变换
fourier_transform = np.fft.fft(signal)
# 计算频率轴
freqs = np.fft.fftfreq(len(signal), d=1/len(signal))
# 输出结果
print("Fourier Transform:", fourier_transform)
print("Frequency Axis:", freqs)
在这个例子中,我们首先导入了numpy库并创建了一个简单的正弦波信号。然后,我们使用np.fft.fft()
函数对信号进行傅里叶变换。最后,我们计算了频率轴并使用print()
函数输出了结果。
注意:np.fft.fft()
函数返回的结果是一个复数数组,它包含了信号的实部和虚部。如果你只关心实部,可以使用np.real()
函数提取出来。同样,如果你只关心虚部,可以使用np.imag()
函数提取出来。