How to transfer a sparse matrix into a block diagnal matrix efficiently?

Question

Ming 2013-4-23

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/73310-how-to-transfer-a-sparse-matrix-into-a-block-diagnal-matrix-efficiently

Hi, Everyone:

Suppose I have a very large M*N sparse matrix A, where M=K*N, I need to equally split it into K N*N matrices and put all of them on the diagnal of a large matrix,i.e.:

 [N*N(1) 0 0 0...0 0 0 ;
0 N*N(2) 0 0 ... 0 ;
0    .   0 0 ... 0 ;
0 0 0  . 0 0 ... 0 ;
0 0 0 0  . 0 ... 0 ;
0 ... 0 0 0 0 N*N(K)]'

I can't use loop because K is very big, so I tried to use:

 B=mat2cell(A,N*ones(K,1),N);       
 C=[blkdiag(B{1:end,1})];

But I found mat2cell is still quite slow when K is very big, is there any other more efficient way to do this?

Many Thanks

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Cedric 2013-4-23

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/73310-how-to-transfer-a-sparse-matrix-into-a-block-diagnal-matrix-efficiently#answer_83258

编辑：Cedric 2013-4-23

在 MATLAB Online 中打开

I would go for something like:

 % - Build example case.
 K = 3 ;
 N = 4 ; 
 A = sparse(randi(10, K*N, N)-1) ;
 % - Extract rows,cols,vals, and reindex columns.
 [r,c,v] = find(A) ;
 c = c + floor((r-1)/N) * N ;
 % - Build block diagonal version.
 B = sparse(r, c, v, K*N, K*N)

With this, you have:

 >> full(A)
 ans =
   9     6     6
   4     7     3
   8     7     9
   1     3     0
   4     6     4
   9     1     3
   7     7     7
   9     0     7
   6     2     1
   0     0     4
   8     0     4
   9     8     6
 >> full(B)
 ans =
   9     6     6     0     0     0     0     0     0     0     0
   4     7     3     0     0     0     0     0     0     0     0
   8     7     9     0     0     0     0     0     0     0     0
   1     3     0     0     0     0     0     0     0     0     0
   0     0     0     6     4     6     4     0     0     0     0
   0     0     0     0     9     1     3     0     0     0     0
   0     0     0     2     7     7     7     0     0     0     0
   0     0     0     5     9     0     7     0     0     0     0
   0     0     0     0     0     0     0     9     6     2     1
   0     0     0     0     0     0     0     9     0     0     4
   0     0     0     0     0     0     0     1     8     0     4
   0     0     0     0     0     0     0     9     9     8     6