写一个二维music谱峰搜索的matlab程序

二维MUSIC算法的谱峰搜索可以通过MATLAB中的“findpeaks”函数实现。下面是一个简单的二维MUSIC算法的谱峰搜索MATLAB程序：

% 二维MUSIC算法谱峰搜索
% Pmusic: 二维MUSIC算法的输出谱
% K: 需要搜索的谱峰数量
% threshold: 谱峰的阈值
function [theta, Pmusic_peaks] = music_2d_peak_search(Pmusic, K, threshold)
    [pks, locs] = findpeaks(Pmusic);  % 找到所有的峰值和对应的位置
    [~, idx] = sort(pks, 'descend');  % 对所有的峰值进行排序
    num_peaks = min(K, length(pks));  % 取前K个最大的峰值
    Pmusic_peaks = zeros(num_peaks, 1);  % 初始化输出谱峰
    theta = zeros(num_peaks, 1);  % 初始化输出角度
    for ii = 1:num_peaks
        if pks(idx(ii)) >= threshold
            Pmusic_peaks(ii) = pks(idx(ii));  % 记录谱峰值
            theta(ii) = locs(idx(ii));  % 记录谱峰对应的角度
        else
            break;
        end
    end
end

其中，输入参数Pmusic为二维MUSIC算法的输出谱，K为需要搜索的谱峰数量，threshold为谱峰的阈值。输出参数theta为搜索得到的谱峰对应的角度，Pmusic_peaks为对应的谱峰值。