Vectorizing Code - how can I populate elements of a matrix from a vector without a FOR loop?

Question

Eric 2011-9-26

1
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/16766-vectorizing-code-how-can-i-populate-elements-of-a-matrix-from-a-vector-without-a-for-loop

I need to populate a large matrix from elements of a vector without using a FOR loop. The sizes of my vector and matrix can be quite large but the problem has the form of this shortened version:

a = [a1 a2 a3]
B = [a1 a1 0 0 0 0;...
  a1 a1 0 0 0 0;...
  0 0 a2 a2 0 0;...
  0 0 a2 a2 0 0;...
  0 0 0 0 a3 a3;...
  0 0 0 0 a3 a3]

This is for a time simulation where the elements of "a" change every time step. The form of "B" does not change, only the elements from "a".

Again, this is easily done with a FOR loop, but in the interest of speed, I really need to avoid using loops here. I'm hoping there is a simple answer but for now, I'm stumped.

Thoughts and suggestions please!

Cheers,

- Eric

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Jan 2011-9-26

Please post the FOR loop method to allow a speed comparison.

请先登录，再进行评论。

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

Answer 1

Fangjun Jiang 2011-9-26

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/16766-vectorizing-code-how-can-i-populate-elements-of-a-matrix-from-a-vector-without-a-for-loop#answer_22618

在 MATLAB Online 中打开

a=[1 2 3];
B=arrayfun(@(x) x*ones(2),a,'uni',0);
B=blkdiag(B{:})

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

请先登录，再进行评论。

Answer 2

Jan 2011-9-26

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/16766-vectorizing-code-how-can-i-populate-elements-of-a-matrix-from-a-vector-without-a-for-loop#answer_22622

在 MATLAB Online 中打开

a = rand(1, 3);
B = kron(diag(a), ones(2));

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

请先登录，再进行评论。

Answer 3

David Holdaway 2012-5-29

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/16766-vectorizing-code-how-can-i-populate-elements-of-a-matrix-from-a-vector-without-a-for-loop#answer_49381

Do you NEED to make B as a matrix? Fangjun Jiang and Jan Simon's methods will work fine for what you have asked, but I've found if your vectors get big the ordinary "*" multiplication in Matlab is not very efficient for large block-diagonal matrices and it takes masses of memory to store and you are better off looping over the blocks anyway. (This is assuming there are no clever inbuilt tools in Matlab for block diagonal matrices I have never heard of).