How to create a loop to sum across columns conditional on index matching?

1 ビュー (過去 30 日間)
Matlabio
Matlabio 2016 年 10 月 25 日
コメント済み: Matlabio 2016 年 11 月 3 日
Hello, I have a binary matrix, with a corresponding index in the first column. For example:
A = [1 1 0 1 0 0 1 ..0]
1 0 0 0 1 1 0 ..1
2 ...
2 ...
2 ...
3 ...
[... ... ]
I want to sum all binary elements corresponding to 1 in the index, 2 in the index, etc. (i.e. index is =1 in the first and second row,so sum across each column, and so on). For the index, the number of corresponding rows with same index number varies across the dataset. I would appreciate all help and advice on how to do this.
  2 件のコメント
Matlabio
Matlabio 2016 年 10 月 26 日
I actually want to have a conditional sum - if index values match in any 2 or more rows, then sum each column in these rows to create a new summation vector. For instance: assume I have the following data matrix (with index in the 1st column): A = [1 1 0 0 0; 2 0 1 0 0; 2 0 0 0 1] In this case, for i =1, my result would be: 1 0 0 0; for i=2 => 0 1 0 1.

サインインしてコメントする。

採用された回答

Chris Dischner
Chris Dischner 2016 年 10 月 25 日
What about something like this?
Assuming your index values are sequential:
aMax = max(A(:,1));
for i = 1:aMax
aSum(i,:) = sum(A(A(:,1) == i,:));
end
Will get you the sums. You can use modular arithmetic to get you to the bit representation
(eg, 1+1+1 = 3 ==> 3 mod 2 = 1
  2 件のコメント
Matt J
Matt J 2016 年 10 月 27 日
編集済み: Matt J 2016 年 10 月 27 日
I get the summation vector for each i with some values >1, which should not be the case since it's strictly binary and cannot exceed 1 in any column sum, for any i
The chances are excellent that it is the input data's fault and not the code's, i.e, that you have not successfully fulfilled the condition "cannot exceed 1 in any column sum, for any i". There is no reason this should be the case just because A(:,2:end) are all zeros and ones.

サインインしてコメントする。

その他の回答 (2 件)

Matt J
Matt J 2016 年 10 月 27 日
編集済み: Matt J 2016 年 10 月 27 日
The following produces a result B, such that B(1,:) is the consolidation of all rows of A with index 1, B(2,:) all rows with index 2, etc...
m=size(A,1);
p=max(A(:,1));
B=sparse(A(:,1),1:m,1,p,m)*A(:,2:end);
  1 件のコメント
Matlabio
Matlabio 2016 年 11 月 3 日
Thank you so much! You were actually right - there was an issue with the data which I have corrected, otherwise it worked perfectly.

サインインしてコメントする。


Teja Muppirala
Teja Muppirala 2016 年 10 月 28 日
result = [unique(A(:,1)) grpstats(A(:,2:end),A(:,1),@any)] % Keep it binary
or
result = [unique(A(:,1)) grpstats(A(:,2:end),A(:,1),@sum)] % Take the sum

タグ

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by