interesting matrix indexing question without for loops

Question

Chris Hooper 2018-10-2

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/421903-interesting-matrix-indexing-question-without-for-loops

评论： Chris Hooper 2018-10-3

size(matrixA)=[b,n,m]
size(matrixB)=[n,m]

how can I create

matrixC(i,j)=matrixA(matrixB(i,j),i,j)

without using for-loops? What is this kind of indexing called? Thanks!

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

请先登录，再进行评论。

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

Answer 1

Stephen23 2018-10-3

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/421903-interesting-matrix-indexing-question-without-for-loops#answer_339487

编辑：Stephen23 2018-10-3

在 MATLAB Online 中打开

Use sub2ind to get the linear indices:

A = reshape(1:6*3*2,6,3,2);
B = [6,1;4,3,;5,2];
% Solution with loops:
for ii = 1:size(B,1)
    for jj = 1:size(B,2)
        C(ii,jj) = A(B(ii,jj),ii,jj);
    end
end
% Solution with SUB2IND:
S = size(B);
[I,J] = ndgrid(1:S(1),1:S(2));
X = sub2ind(size(A),B,I,J);
D = A(X)
% Compare:
isequal(C,D)