RLS自适应陷波器的matlab代码

下面是一个简单的 RLS 自适应陷波器的 Matlab 代码：

% 参数设置
N = 1000; % 信号长度
f0 = 0.1; % 信号频率
theta = 2*pi*f0; % 信号角频率
fs = 1000; % 采样频率
t = (0:N-1)/fs; % 时间向量
x = sin(theta*t); % 信号

% 添加干扰信号
f1 = 0.2; % 干扰信号频率
theta1 = 2*pi*f1; % 干扰信号角频率
d = sin(theta*t) + 0.5*sin(theta1*t); % 输入信号+干扰信号

% RLS 自适应陷波器
M = 51; % 陷波器长度
lambda = 0.99; % 遗忘因子
delta = 0.01; % 初始误差方差
w = zeros(M,1); % 初始权值
P = eye(M)/delta; % 初始协方差矩阵
e = zeros(N,1); % 存储误差信号
for n = 1:N
    u = [x(n:-1:n-M+1);x(n+1:M)]; % 输入信号
    y = w'*u; % 陷波器输出
    e(n) = d(n) - y; % 计算误差信号
    k = P*u/(lambda+u'*P*u); % 计算增益
    w = w + k*e(n); % 更新权值
    P = (P-k*u'*P)/lambda; % 更新协方差矩阵
end

% 绘图
figure;
subplot(3,1,1);
plot(t,x); % 原始信号
xlabel('Time (s)');
ylabel('Amplitude');
title('Original Signal');

subplot(3,1,2);
plot(t,d); % 输入信号+干扰信号
xlabel('Time (s)');
ylabel('Amplitude');
title('Input Signal + Interference');

subplot(3,1,3);
plot(t,e); % 误差信号
xlabel('Time (s)');
ylabel('Amplitude');
title('Error Signal');