How to create a block diagonal matrix from the slices of a 3D matrix
Show older comments
I have a large 3D matrix (K) composed of 200 slices , and I want to create a block diagonal matrix from the slices of it without looping. For example
K(:,:,1) =[1 2;3 4]
K(:,:,2) =[5 6;7 8]
K(:,:,3) =[9 10;11 12]
.
.
.
K(:,:,200) =[1 10;13 15]
if we assume the number of slice is 3, the output should be
1 2 0 0 0 0
3 4 0 0 0 0
0 0 5 6 0 0
0 0 7 8 0 0
0 0 0 0 9 10
0 0 0 0 11 12
But how the block diagonal matrix created from all the 200 slices without looping?
Accepted Answer
More Answers (3)
I doubt that avoiding loops leads to the fastest code, but the following does avoid them,
[m,n,p]=size(K);
BlockMatrix=kron(speye(p), ones(m,n));
BlockMatrix(logical(BlockMatrix))=K(:);
This doesn't avoid a loop, even if it looks like it does, but it might be the fastest way.
Kcell = cellfun(@sparse, num2cell(K,[1,2]), 'uni',0 );
BlockMatrix=blkdiag(Kcell{:});
1 Comment
Yunsong Liu
on 18 Jul 2019
Thanks Matt. It works very well.
Azzi Abdelmalek
on 2 Oct 2014
k(:,:,1) =[1 2;3 4]
k(:,:,2) =[5 6;7 8]
k(:,:,3) =[9 10;11 12]
out=[];
for ii=1:size(k,3)
out=blkdiag(out,k(:,:,ii))
end
Categories
Find more on Operating on Diagonal Matrices 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!