Matlab数组中唯一元素的累积计数

Question

使用 Matlab 2019b。

x = [10 10 10 20 20 30]';

如何获得 x 中唯一元素的累积计数，应该如下所示：

y = [1 2 3 1 2 1]';

编辑：

我的真实数组实际上比上面给出的示例要长得多。以下是我测试的方法：

x = randi([1 100], 100000, 1);
x = sort(x);

% method 1: check neighboring values in one loop
tic
y = ones(size(x));
for ii = 2:length(x)
    if x(ii) == x(ii-1)
        y(ii) = y(ii-1) + 1;
    end
end
toc

% method 2 (Wolfie): count occurrence of unique values explicitly
tic
u = unique(x);
y = zeros(size(x));
for ii = 1:numel(u)
    idx = (x == u(ii));
    y(idx) = 1:nnz(idx);
end
toc

% method 3 (Luis Mendo): triangular matrix
tic
y = sum(triu(x==x'))';
toc

结果：

Method 1: Elapsed time is 0.016847 seconds.
Method 2: Elapsed time is 0.037124 seconds.
Method 3: Elapsed time is 10.350002 seconds.

Answer 1

您可以遍历唯一元素，并将它们的索引设置为 1:n 每次...

u = unique(x);
y = zeros(size(x));
for ii = 1:numel(u)
    idx = (x == u(ii));
    y(idx) = 1:nnz(idx);
end

Answer 2

这有点低效，因为它会生成一个中间矩阵，而实际上只需要一个三角形的一半：

y = sum(triu(x==x.')).';

Answer 3

假设x已排序：

x = [10 10 10 20 20 30].';
x = sort(x);

[~, ic] = cummax(x);
y = (2 : numel(x) + 1).' - ic;

当x 未排序时使用：

[s, is] = sort(x);
[~, ic] = cummax(s);
y(is) = (2 : numel(s) + 1).' - ic;