如何使用 scipy FFT 限制頻率范圍

Question

我正在使用 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 范圍內的頻率，而不是 plot 中看到的 0-15？

Answer 1

您可以使用xlim的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()

plt.plot(xf, 2.0/N * np.abs(yf[0:N//2]),'b')
plt.xlim(0,100) # you need this
plt.grid()
plt.show()