用matlab写esprit算法

以下是用MATLAB编写ESPRIT算法的示例代码：

% 生成具有5个成分的复数信号
N = 1000; % 数据点数
n = 5; % 成分数
A = [1+1i, 1-1i, 2+2i, 2-2i, 3+3i]; % 成分振幅
f = [0.1, 0.2, 0.3, 0.4, 0.5]; % 成分频率
phi = [pi/4, pi/3, pi/2, pi/6, pi/8]; % 成分相位
t = 0:1/N:(N-1)/N;
x = A * exp(1i*2*pi*f'*t) .* exp(1i*phi');

% ESPRIT算法
L = 3; % 阶数
K = n - L; % 信号子空间维数
X = x(:);
M = length(X);
Rxx = toeplitz(conj(X(1:M)), X(1:L));
[U, ~, ~] = svd(Rxx);
U1 = U(:, 1:K);
U2 = U(:, K+1:end);
Ur = U1'*U2;
theta = angle(eig(Ur));
f_esprit = sort(theta/(2*pi));

% 绘制结果
figure;
subplot(211);
plot(t, real(x));
xlabel('时间 (s)');
ylabel('幅值');
title('原始信号');
subplot(212);
stem(f_esprit, ones(1, L));
xlim([0, 0.6]);
xlabel('频率 (Hz)');
ylabel('幅值');
title('ESPRIT估计的频率');

运行代码将生成一个具有5个成分的复数信号，并使用ESPRIT算法估计其频率。结果显示原始信号的频率和ESPRIT估计的频率。