Использование 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
будучи дескриптором функции по умолчанию, вы можете оставить его для этого случая.