Использование accumarray для суммирования данных в Matlab

У меня есть такая матрица:

>>D=[1,0,10;3,1,12;3,1,12.5;6,1,6;6,2,11.1;]
D =

1.0000         0   10.0000
3.0000    1.0000   12.0000
3.0000    1.0000   12.5000
6.0000    1.0000    6.0000
6.0000    2.0000   11.1000

Я хочу получить сумму второго столбца данных, если их первый столбец совпадают. Например, я хочу иметь:

E=
1.0000         0
3.0000    2.0000
6.0000    3.0000

Так я попробовал

b = accumarray(D(:,1),D(:,2),[],[],[],true);
[i,~,v] = find(b);
E = [i,v]

но это не сработало. Что я должен делать?

1 ответ

Решение

Используйте комбинацию unique а также accumarray сюда -

[unique_ids,~,idmatch_indx] = unique(D(:,1)); 
%// unique_ids would have the unique numbers from first column and only 
%// used to get the first column of final output, E. 
%// idmatch_indx are tags put on each element corresponding to each unique_ids
%// based on the uniqueness

%// Accumulate and perform summation of elements from second column of D using
%// subscripts from idmatch_indx
E = [unique_ids accumarray(idmatch_indx,D(:,2))]

С accumarray как правило, вы должны ввести функцию, которую вы хотели бы использовать на накопленные элементы, но @sum будучи дескриптором функции по умолчанию, вы можете оставить его для этого случая.

Другие вопросы по тегам