Python 功率谱估计直接法

最新推荐文章于 2023-09-17 15:10:51 发布

吃烩菜的马卡龙

最新推荐文章于 2023-09-17 15:10:51 发布

阅读量1k

点赞数

分类专栏：笔记文章标签： python 数字信号处理

本文链接：https://blog.csdn.net/tsemzy/article/details/120446113

版权

xn=10sin(2pi0.1+pi/3)+4sin(2pi0.4+pi/4)，噪声为均值为零方差为一的高斯白噪声，用经典谱估计直接法来估计f1（0.1）和f2（0.4），并计算均方误差（MSE）。

原理

对N点DFT取绝对值后得到幅频特性，取绝对值平方后得到功率谱。
在估计f1、f2时，采取对幅频特性拟合后根据波峰处值最大来得到所要估计的f对应的k。
将整个程序循环多次即可计算估计值的期望和均方误差。
如何调整拟合效果见上篇博文。

结果

功率谱

期望与均方误差

代码

import matplotlib.pyplot as plt
import numpy as np
from math import *
import random as ran
#定义Wn
def WN(N,k,n):
    return complex(cos(-2*pi/N*k*n), sin(-2*pi/N*k*n))
#整个程序循环多次来计算MSE
f_1 = 0
f_2 = 0
Error_f1 = 0
Error_f2 = 0
MSE_f1 = 0
MSE_f2 = 0
times = 5
for time in range(0, int(times)):