如何使用 scipy FFT

如何解决如何使用 scipy FFT

我正在使用 FFT 来查找信号的频率。我只对 1 到 4 Hz 之间的特定频率范围感兴趣。

我有这个代码来计算频率：

from scipy.fft import rfft,rfftfreq,irfft

plt.plot(d)
plt.show()
N = len(d)
yf = rfft(d)
xf = rfftfreq(N,1 / sample_rate) # 29
plt.plot(xf,np.abs(yf))
plt.show()

结果：

如何修改我的代码，使 xf 和 yf 仅对应于我想要的 1-4 Hz 范围内的频率，而不是图中看到的 0-15？

解决方法

您可以使用xlim的{​​{1}}特性来修改matplotlib。

这是您可以参考的示例代码。

x axis

from scipy.fft import fft,fftfreq
import numpy as np
# Number of sample points
N = 600
# sample spacing
T = 1.0 / 800.0
x = np.linspace(0.0,N*T,N,endpoint=False)
y = np.sin(50.0 * 2.0*np.pi*x) + 0.5*np.sin(80.0 * 2.0*np.pi*x)
yf = fft(y)
xf = fftfreq(N,T)[:N//2]
import matplotlib.pyplot as plt
plt.plot(xf,2.0/N * np.abs(yf[0:N//2]),'b')
plt.plot()
plt.grid()
plt.show()