清空筛选条件
清空筛选条件

What is the best way to vectorize this matrix problem?

1 次查看（过去 30 天）
John Malik
John Malik 2017-4-27
编辑： Andrei Bobrov 2017-4-28
I have a square matrix A which I would like to manipulate using a "map" matrix D. The result will be the matrix B. The entry B(i, j) will be a sum of a number of elements from A(:, j), as determined by D. In the following example, the matrix D tells me that A(1, 1) will be added with A(3, 1) to form B(1, 1), and so on. When the matrices are large, this process takes a long time. Is there a way to vectorize this computation?
A = magic(3); % initial matrix
D = [1, 2, 3; 2, 2, 3; 1, 3, 1]; % map matrix
B = zeros(3); % resultant matrix
for i = 1:3
for j = 1:3
B(D(i, j), j) = B(D(i, j), j) + A(i, j);
end
end

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

采纳的回答

John Malik
John Malik 2017-4-27
This question has been answered here.

更多回答（1 个）

Andrei Bobrov
Andrei Bobrov 2017-4-27
编辑：Andrei Bobrov 2017-4-28
B = accumarray([D(:),repelem((1:size(D,2))',size(D,1))],A(:));

类别

在 Help Center 和 File Exchange 中查找有关 Resizing and Reshaping Matrices 的更多信息

标签

产品

Community Treasure Hunt

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

Start Hunting!

Translated by