python - 在 Pandas 数据帧中存储 FFT 周期时出现奇怪的问题

Question

我正在尝试将 FFT 计算的结果存储在 Pandas 数据框中:

ft = pd.DataFrame(index=range(90))
ft['y'] = ft.index.map(lambda x: np.sin(2*x))
ft['spectrum']  = np.fft.fft(ft['y'])
ft['freq']      = np.fft.fftfreq(len(ft.index)).real
ft['T']         = ft['freq'].apply(lambda f: 1/f if f != 0 else 0)

一切似乎都工作正常，直到最后一行:应该存储周期的列 T 由于某种原因具有框架的所有列，即:

In [499]: ft.T[0]
Out[499]:
y                            0j
spectrum    (0.913756021471+0j)
freq                         0j
T                            0j
Name: 0, dtype: complex128

我不明白这是为什么。当我只取频率的实部时也会发生这种情况:

ft['freq']  = np.fft.fftfreq(len(ft.index)).real

或者我尝试使用替代方法计算 T 值，例如:

ft.T = ft.index.map(lambda i: 1/ft.freq[i] if ft.freq[i] else np.inf)
ft.T = 1/ft.freq

当我对它们运行 head() 或 describe() 时，所有其他列看起来都很整洁，无论它们包含实数还是复数值。 freq 列看起来像一个普通的一维序列，因为 np.fft.fftfreq() 返回一维复数数组，那么 T 列如此困惑的原因可能是什么？

我正在使用 Pandas v.1.19.2 和 Numpy v.1.12.0。

Answer 1

Pandas DataFrame 对象有 a property称为T，用于“转置DataFrame对象的索引和列”。如果您使用不同的列名称而不是 T，一切都会按预期工作。