傅立叶变换(np.fft)结果中出现意外相移

时间:2020-10-06 10:34:57

标签: python numpy fft

我正在寻求澄清傅立叶变换原理。

我尝试做一些非常简单的事情:创建一个信号(具有给定频率和相移的正弦波),并通过傅立叶变换重新创建其参数。频率估算工作正常,但在相位方面,好像我得到系统的偏移(-pi / 2)。

import numpy as np
import matplotlib.pyplot as plt

duration = 4.0 # lenght of window (in sec)
ticks_per_sec = 400.0 # sampling interval 
samples = int(ticks_per_sec*duration)

phase_shift = np.pi / 2 # sin wave shift in angle
freq = 1 # sine wave freq in Hz

phase_shift  = round(ticks_per_sec/freq * phase_shift/(2*np.pi)) # angle value translated to no of ticks
t = np.arange(phase_shift, samples+phase_shift) / ticks_per_sec

s =  1 * np.sin(2.0 * np.pi * freq * t)
N = s.size

fig, axs = plt.subplots(1, 3, figsize=(18, 6))
axs[0].grid(True)
axs[0].set_ylabel("Amplitude")
axs[0].set_xlabel("Time [s]")
axs[0].set_title(f"F: {freq}, Phase shift: {phase_shift} ticks.")
axs[0].plot(np.arange(samples)/ticks_per_sec, s) 

f = np.linspace(0, ticks_per_sec, N)
fft = np.fft.fft(s)
peak_pos = np.argmax(np.abs(fft[:N//2])) 

axs[1].set_ylabel("Amplitude")
axs[1].set_xlabel("Frequency [Hz]")
axs[1].set_title(f"Peak bar: {peak_pos}")
barlist = axs[1].bar(f[:N // 2], np.abs(fft)[:N // 2] * (1 / (N//2)), width=1.5)  # 1 / N is a normalization factor
barlist[peak_pos].set_color('r')


axs[2].set_ylabel("Angle")
axs[2].set_xlabel("Frequency [Hz]")
axs[2].set_title(f"Peak angle: {np.angle(fft[peak_pos])}")
barlist = axs[2].bar(f[:N // 2], np.angle(fft)[:N // 2], width=1.5) 
barlist[peak_pos].set_color('r')

fig.show()

Plotted Results of the code above

如果我的代码中有一个我不注意到的错误,或者我误会了某些东西,请帮助我。

谢谢。

1 个答案:

答案 0 :(得分:1)

您的代码很好,这不是编程问题。

让我们回想一下,正弦波可以表示为具有相移的余弦波(反之亦然),现在请记住,相对于余弦,在真正的傅立叶基础上,正弦函数具有-pi / 2的固有相移。 / p>

这意味着当将np.sin替换为np.cos时,您的代码应输出pi / 2相角, ie 返回输入phase_shift,或者等效地,返回指定phase_shift = np.pi / 2 ie phase shift和正弦相位时,相位角为零。