在Python中确定数组的频率

Question

我有一个充满浮点数的示例文件，如下所示：

    -0.02  3.04  3.04  3.02  3.02  3.06  3.04  3.02  3.04  3.02  3.04  3.02
     3.04  3.02  3.04  3.04  3.04  3.02  3.04  3.02  3.04  3.02  3.04  3.02
     3.06  3.02  3.04  3.02  3.04  3.02  3.02  3.06  3.04  3.02  3.04  3.02
     3.04  3.02  3.04  3.04  3.04  3.02  3.04  3.02  3.02  3.06  3.04  3.02
     3.06  3.02  3.04 -0.02 -0.02 -0.02 -0.02 -0.02 -0.02 -0.04 -0.02 -0.04

这些数字放置在文本文件中。我正在尝试读取文本文件并确定该信号的频率。该数据是从数字示波器捕获的。我可以在示波器显示中看到频率，但我还想通过在 Python 中处理它来验证它。我在PC端使用Python从设备捕获数据。

尽管我可以用 Python 做一些低级的事情，但我对文本处理完全是新手。我想我需要首先将文件中的数据加载到数组中，然后执行 FFT 或更简单的算法，生成以 Hz 为单位的整数。

理论上，我知道如何进行傅里叶分析，并且我可以在纸上对任何给定的信号进行分析。对于给定的数据集，我不知道从哪里开始使用 Python。我已经尝试过 scipy-numpy 的文档，但对我来说效果不太好。

我希望得到有经验的用户的指导。

Answer 1

从您的问题中不清楚文件中的值到底代表什么。但假设它们指示连续的电压样本，您可以使用将文件加载到 Numpy 数组中

import numpy as np
data = np.array([float(f) for f in file(filename).read().split()])

，然后计算傅里叶变换，

import numpy.fft as fft
spectrum = fft.fft(data)

然后您可以绘制 FFT 的幅度，

freq = fft.fftfreq(len(spectrum))
plot(freq, abs(spectrum))

并且您看到的内容应与示波器上显示的内容相匹配。

如果您想识别频谱中的主要频率，则必须在某个阈值处切割数组，例如：

threshold = 0.5 * max(abs(spectrum))
mask = abs(spectrum) > threshold
peaks = freq[mask]

freq 的内容（因此还有 peaks code>) 是以采样率为单位的频率。例如，如果您的示波器每微秒采样一次波形，则 freq 中的值以兆赫为单位。因此，如果您输入理想的 1 kHz 信号（例如，

t = arange(4e6) / 1e6 # sampling times in seconds
data = sin(2 * pi * 1000 * t)

您将在 0.001 MHz 处获得峰值），因此您会发现 peaks = array([-0.001, 0.001]) 。