Counting the same rows of 2D matrix

2019-02-25 04:09发布

I have a two column matrix. I need to make it three column, where the third column shows the number of appearance of the first two as a row in the input matrix.

Basically: Input

[1 1;
 1 1;
 1 2;
 1 2;
 1 3]

Desired output:

[1 1 2;
 1 2 2;
 1 3 1]

I know already that a proper combination of accumarray and unique should do the charm. I just don't know how to properly combine them.

2条回答
男人必须洒脱
2楼-- · 2019-02-25 04:37

One possible solution:

clear
a=...
[1 1;
 1 1;
 1 2;
 1 2;
 1 3]

[U,~,ic]=unique(a,'rows');
[C] = histc(ic,unique(ic));
Result=[U,C]
查看更多
相关推荐>>
3楼-- · 2019-02-25 04:41

You are right, unique and accumarray are perfectly suited for this task:

x = [1 1; 1 1; 1 2; 1 2; 1 3]; % input
[~, v, w] = unique(x, 'rows', 'stable'); % unique indices and labels
c = accumarray(w, 1); % counts
y = [x(v,:) c]; % output

Remove the 'stable' flag if you want the ouput rows sorted in lexicographical order.

You can also replace accumarray by bsxfun to obtain the counts:

c = sum(bsxfun(@eq, unique(w), w.'), 2);

For the special case that the entries of x are positive integers and you want the ouput in lexicographical order, you can also use sparse and find as follows:

x = [1 1; 1 1; 1 2; 1 2; 1 3]; % input
[ii,jj,vv] = find(sparse(x(:,1), x(:,2), 1));
y = [ii(:), jj(:), vv(:)]; % output
查看更多
登录 后发表回答