matlab中函数D = pdist2(X,Y,distance,'Smallest',K)代码编写

Question

求matlab函数pdist2中，D = pdist2(X,Y,distance,'Smallest',K)，的代码编写，用matlab编写这个函数

Answer 1

计算两组 n × p 和 m × p 维数据之间所有可能的二次距离，返回前 K 个最小距离和对应的索引。

function [D,I] = pdist2(X,Y,distance,~,K)
% Calculate pairwise distances between two sets of vectors
% X: n x p matrix, Y: m x p matrix, distance: distance metric string,
% K: number of smallest distances to return

n = size(X,1);
m = size(Y,1);

switch distance
    case 'euclidean'
        % Euclidean distance
        D = sqrt(sum(X.^2,2) + sum(Y.^2,2).' - 2*X*Y.');
    case 'sqeuclidean'
        % Squared Euclidean distance
        D = sum(X.^2,2) + sum(Y.^2,2).' - 2*X*Y.';
    case 'cityblock'
        % City block distance
        D = abs(bsxfun(@minus,X,permute(Y,[3 2 1])));
        D = permute(sum(D,2),[1 3 2]);
    case 'cosine'
        % Cosine distance
        D = 1 - (X*Y.' ./ (sqrt(sum(X.^2,2))*sqrt(sum(Y.^2,2).')));
    otherwise
        error('Unknown distance metric');
end

% Sort distances and return smallest K
[D,idx] = sort(D,2,'ascend');
D = D(:,1:K);
I = zeros(n,K);
for i = 1:n
    I(i,:) = idx(i,1:K);
end

end

这个函数首先读入 X 和 Y 两个输入矩阵，计算它们之间的距离，具体方法根据输入变量 distance 而有所不同。距离矩阵 D 然后按升序排序，并返回前 K 个最小值和对应的索引。