我有Velocity与时间的数据。时间步长不均匀,但速度数据是波形。如何使用Python的FFT计算速度的主频率?我在网上看到的大多数例子都是为了统一时间步进。
我的数据就像
7.56683038E+02 2.12072850E-01
7.56703750E+02 2.13280844E-01
7.56724461E+02 2.14506402E-01
7.56745172E+02 2.15748934E-01
7.56765884E+02 2.17007907E-01
7.56786595E+02 2.18282753E-01
10000条线。
看到一些在线回复,我写了一个如下代码,但它不起作用:
#!/usr/bin/env python
import numpy as np
import scipy as sy
import scipy.fftpack as syfp
import pylab as pyl
# Calculate the number of data points
length = 0
for line in open("data.dat"):
length = length + 1
# Read in data from file here
t = np.zeros(shape=(length,1))
u = np.zeros(shape=(length,1))
length = 0
for line in open("data.dat"):
columns = line.split(' ')
t[length] = float(columns[0])
u[length] = float(columns[1])
length = length + 1
# Do FFT analysis of array
FFT = sy.fft(u)
# Getting the related frequencies
freqs = syfp.fftfreq(len(u))
# Create subplot windows and show plot
pyl.subplot(211)
pyl.plot(t, u)
pyl.xlabel('Time')
pyl.ylabel('Amplitude')
pyl.subplot(212)
pyl.plot(freqs, sy.log10(FFT), 'x')
pyl.show()
----------------------编辑------------------------
使用此代码我得到如下图所示的输出。我不确定这个数字是什么。我期待在FFT图中看到一个峰值
----------------------编辑------------------------
我在以下评论中建议的带有sin函数的模拟数据的结果如下:
答案 0 :(得分:5)
从我所看到的,你的代码基本上很好,但缺少一些细节。我认为你的问题主要是解释。因此,模拟数据现在是最好看的,这里是我在评论中建议的模拟数据的示例(我添加了关于重要行的注释,并##
进行了更改):
import numpy as np
import scipy as sy
import scipy.fftpack as syfp
import pylab as pyl
dt = 0.02071
t = dt*np.arange(100000) ## time at the same spacing of your data
u = np.sin(2*np.pi*t*.01) ## Note freq=0.01 Hz
# Do FFT analysis of array
FFT = sy.fft(u)
# Getting the related frequencies
freqs = syfp.fftfreq(len(u), dt) ## added dt, so x-axis is in meaningful units
# Create subplot windows and show plot
pyl.subplot(211)
pyl.plot(t, u)
pyl.xlabel('Time')
pyl.ylabel('Amplitude')
pyl.subplot(212)
pyl.plot(freqs, sy.log10(abs(FFT)), '.') ## it's important to have the abs here
pyl.xlim(-.05, .05) ## zoom into see what's going on at the peak
pyl.show()
如您所见,正如预期的那样,在+和 - 输入频率(.01 Hz)处有两个峰值。
修改强>:
很困惑为什么这种方法对于OP的数据不起作用,我也看了一下。问题是样本时间间隔不均匀。这是时间的直方图(下面的代码)。
因此,样本之间的时间在短时间和长时间之间大致均匀分配。我在这里快速查看了一个模式,没有什么是显而易见的。
要进行FFT,需要均匀间隔的时间样本,因此我进行了内插以获得以下内容:
这是合理的(DC偏移,主峰和小谐波)。这是代码:
data = np.loadtxt("data.dat", usecols=(0,1))
t = data[:,0]
u = data[:,1]
tdiff = t[1:]-t[:-1]
plt.hist(tdiff)
new_times = np.linspace(min(t), max(t), len(t))
new_data = np.interp(new_times, t, u)
t, u = new_times, new_data
FFT = sy.fft(u)
freqs = syfp.fftfreq(len(u), dt)
# then plot as above