Sum: every nth column and groups/blocks of columns

2 ビュー (過去 30 日間)
Dom
Dom 2022 年 8 月 24 日
編集済み: Dyuman Joshi 2022 年 8 月 24 日
Hi:
Is there any way to improve this code?
Thanks
Dominic
%
% Flows between sectors of countries
% 3 countries with 4 sectors
% Row and column headings are the same C1S1 C1S2 ......C4S3 C4S4
%
c=3; % no. of countries
s=4; % no. of sectors
X=magic(c*s); % 12
%
% add every 4th (sth) column of a matrix: 1,5,9 ... 4,8,12
% - result 12x4 (sum by sector)
%
S=sum(X(:,1:s:end),2); % 1:4:end
for k=2:s % 2:4
SS=sum(X(:,(k):s:end),2);
S=horzcat(S,SS);
end
%
% add columns in blocks of 4: 1-4, 5-8, 9-12 - result 12x3 (sum by country)
%
C=sum(X(:,[1:s]),2); % [1:4]
x=(s+1):s:(c*s); % 5:4:12
y=(2*s):s:(c*s); % 8:4:12
for k=[x;y]
CC=sum(X(:,[k(1):k(2)]),2);
C=horzcat(C,CC);
end

採用された回答

Bruno Luong
Bruno Luong 2022 年 8 月 24 日
編集済み: Bruno Luong 2022 年 8 月 24 日
c=3; % no. of countries
s=4; % no. of sectors
X=magic(c*s); % 12
XX = reshape(X,size(X,1),s,[]);
S = sum(XX,3)
S = 12×4
420 18 21 411 51 381 378 60 87 345 342 96 312 126 129 303 276 162 165 267 195 237 234 204 231 201 198 240 168 270 273 159 132 306 309 123 339 93 90 348
C = squeeze(sum(XX,2))
C = 12×3
290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290
  3 件のコメント
Dom
Dom 2022 年 8 月 24 日
Thanks Bruno. I am sure it is.

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

その他の回答 (2 件)

Fangjun Jiang
Fangjun Jiang 2022 年 8 月 24 日
x=magic(12);
y=reshape(x,12,4,[]);
z=squeeze(sum(y,2))
z = 12×3
290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290
  1 件のコメント
Dom
Dom 2022 年 8 月 24 日
Thank you Jiang, very elegant. Any comments on the first part?
Thanks again for taking the time.

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


Dyuman Joshi
Dyuman Joshi 2022 年 8 月 24 日
編集済み: Dyuman Joshi 2022 年 8 月 24 日
%comment lines removed
c=3; % no. of countries
s=4; % no. of sectors
X=magic(c*s); % 12
Vectorized solution
S1=reshape(sum(reshape(X,[],c),2),[],s)
S1 = 12×4
420 18 21 411 51 381 378 60 87 345 342 96 312 126 129 303 276 162 165 267 195 237 234 204 231 201 198 240 168 270 273 159 132 306 309 123 339 93 90 348
C1=reshape(sum(reshape(X',s,[])),[],c)
C1 = 12×3
290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290 290
  3 件のコメント
Dom
Dom 2022 年 8 月 24 日
Thanks Dyuman. I have looked at all three contributions. @Bruno Luong and @Fangjun Jiang add another dimension. I am stuck in old loop and 2x2 thinking. All the answers work so I wonder why I can't accept all of them. Thanks.

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

カテゴリ

Find more on Loops and Conditional Statements in Help Center and File Exchange

タグ

Community Treasure Hunt

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

Start Hunting!

Translated by