如何计算给定波的频率和时间

时间:2014-09-29 17:40:15

标签: python numpy scipy fft dft

我有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图中看到一个峰值 Output by the python program

----------------------编辑------------------------

我在以下评论中建议的带有sin函数的模拟数据的结果如下:

results of the mock data

1 个答案:

答案 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()

enter image description here

如您所见,正如预期的那样,在+和 - 输入频率(.01 Hz)处有两个峰值。

修改
很困惑为什么这种方法对于OP的数据不起作用,我也看了一下。问题是样本时间间隔不均匀。这是时间的直方图(下面的代码)。

enter image description here

因此,样本之间的时间在短时间和长时间之间大致均匀分配。我在这里快速查看了一个模式,没有什么是显而易见的。

要进行FFT,需要均匀间隔的时间样本,因此我进行了内插以获得以下内容:

enter image description here

这是合理的(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