如何将数据输入到 FFT

Question

所以我有两个数据数组：

Bx = [  -59.57011259   -74.20675537   -90.53224156 ..., -1676.9703173
-1676.9703173  -1676.9703173 ]
By = [  1.48413511e+00   4.96417605e+00   8.39303992e+00 ...,  -1.67697032e+03
-1.67697032e+03  -1.67697032e+03]

我如何让它们接收看起来像这样的 FFT？

我有一个程序可以显示这些数据，但我需要在 Python2.7 内完成。我尝试使用我在本主题 (Plotting a Fast Fourier Transform in Python) 中找到的代码，但老实说，我在理解 FFT 方面遇到了困难，你能帮忙吗？

# Number of samples
N = 600
# Sample spacing
T = 300.0 / 266336
x = np.linspace(0.0, N*T, N)
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 = np.linspace(0.0, 1.0/(2.0*T), N/2)
plt.plot(xf, 2.0/N * np.abs(yf[0:N/2]))
plt.grid()
plt.show()

关于我的数据的一些信息：数量 records/samples 266336；时间 300s = 300000ms

我还需要以某种方式实现 blackman 或 hamming window，你能帮忙吗？

Answer 1

假设Bx和By是类数组，开窗可以用*运算符来写。

import numpy as np
import matplotlib.pyplot as plt

# Number of samplepoints
N = 266336
# sample spacing
T = 300.0 / N

# Window
win = np.hamming(N)

x = np.linspace(0.0, N*T, N)
# y = np.sin(50.0 * 2.0*np.pi*x) + 0.5*np.sin(80.0 * 2.0*np.pi*x)
y = np.array(Bx)
y_win = win * y
yf = np.fft.fft(y_win)
xf = np.linspace(0.0, 1.0/(2.0*T), N/2)

# Plot original data
ax = plt.subplot(3,1,1)
ax.grid()
ax.plot(y)

# Plot windowed data
ax = plt.subplot(3,1,2)
ax.grid()
ax.plot(y_win)

# Plot spectrum
ax = plt.subplot(3,1,3)
ax.grid()
ax.plot(xf, 2.0/N * np.abs(yf[:N/2]))
plt.show()

Answer 2

我用过 Welch 方法，这是个好主意。 post 结束。问题已解决。

# Loop for FFT data
for dataset in [fft1]:
    dataset = np.asarray(dataset)
    freqs, psd = welch(dataset, fs=266336/300, window='hamming', nperseg=8192)
    plt.semilogy(freqs, psd/dataset.size**2, color='r')

for dataset2 in [fft2]:
    dataset2 = np.asarray(dataset2)
    freqs2, psd2 = welch(dataset2, fs=266336/300, window='hamming', nperseg=8192)
    plt.semilogy(freqs2, psd2/dataset2.size**2, color='b')

如何将数据输入到 FFT

How to input data to FFT

python

numpy

python-2.7

hamming-window