numpy fft 在绘图时返回原始频率

Question

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

df = pd.read_csv('signal180_single.csv', sep=',', header=None)
x = df.values
length = len(x)
# frequency 0.02s
fs = 50.0
t = np.arange(0, length/fs,1.0/fs)

xF = np.fft.fft(x)
N = len(xF)
xF = xF[0:N/2]

# plot frequencys from 0 to fs, with num = N/2
fr = np.linspace(0,fs,N/2)

plt.figure()
plt.subplot(211)
plt.plot(t,x)
plt.subplot(212)
plt.plot(fr, abs(xF))
plt.show()

我正在将 180000 个浮点值从文件写入数组。 这些值以 50Hz 采样，包含 2Hz 的正弦波。

然后我在上面的绘图窗口中绘制频率。 我想在下部绘图窗口中绘制频率的频谱，但我得到与上部绘图窗口相同的值。 谁能看到错误在哪里？

当我绘制公式 x = np.sin(10*t) + np.cos(3*t) 时，我得到了频率。但不是当我从文件或数组中读取窦性。

Answer 1

一行

x = df.values 

返回一个 180000 x 1 的数组。每个值都单独存储。在这种情况下，将行替换为：

x = df.values.ravel()

您的脚本将起作用。