用python 实现2D FFT

最新推荐文章于 2024-09-26 11:04:00 发布

原创

最新推荐文章于 2024-09-26 11:04:00 发布 · 3.7k 阅读

21 ·

CC 4.0 BY-SA版权

文章标签：

#python #fft #信号处理

该博客介绍了如何利用Python进行2D快速傅里叶变换（FFT），并展示了2DFFT的结果。首先，创建了一个包含多个正弦波的2D矩阵，然后使用`fftn`函数计算2DFFT，并通过`fftshift`进行频率轴调整。最后，通过2D和3D绘图展示频谱，并进行了相应的频率轴标记。

用python实现2D FFT

前言

用python实现2D FFT，并绘制结果

代码

直接上代码

import numpy as np
from scipy.fftpack import fftshift,fftn
import matplotlib.pyplot as plt
from matplotlib.pyplot import MultipleLocator

N = 128                           # 采样点数
sample_freq = 120                 # 采样频率 120 Hz, 大于两倍的最高频率
sample_interval = 1/sample_freq   # 采样间隔
signal_len=N*sample_interval      # 信号长度
t=np.arange(0, signal_len, sample_interval)


signal = 5 + 2 * np.sin(2 * np.pi * 15 * t) + 3 * np