ESPRIT算法matlab

以下是ESPRIT算法的Matlab实现示例：

% 生成信号
N = 1000; % 信号长度
f1 = 100; % 信号频率
f2 = 200;
A1 = 1; % 信号幅值
A2 = 0.5;
n = 0:N-1;
s = A1*sin(2*pi*f1*n/N) + A2*sin(2*pi*f2*n/N);

% ESPRIT算法
p = 2; % 信号个数
L = N/2; % 子空间维数
X = hankel(s(1:L), s(L:N)); % 构造Hankel矩阵
[U, S, V] = svd(X); % 奇异值分解
U1 = U(:, 1:p);
U2 = U(:, p+1:end);
R = U1' * U2; % 计算相关矩阵
[eigvec, eigval] = eig(R);
[~, index] = sort(diag(eigval), 'descend');P = eigvec(:, index(1:p)); % 计算信号子空间
theta = angle(eig(eigval(index(1:p), index(1:p)))); % 估计信号频率
f_est = sort(theta/(2*pi));

% 显示结果
disp(['估计的频率为：', num2str(f_est')]);

% 绘制频谱图
NFFT = 2^nextpow2(N);
Y = fft(s, NFFT)/N;
f = 1/2*linspace(0, 1, NFFT/2+1);
figure;
plot(f, 2*abs(Y(1:NFFT/2+1)));
xlabel('Frequency (Hz)');
ylabel('Amplitude');
title('Single-Sided Amplitude Spectrum of s(t)');
grid on;