如何在matlab中从第三维选择第三矩阵中的特定索引

Question

我知道这是一个简单的问题，但很难用一句话来表达答案。所以，我有一个像这样大小为 2x2x3 的 3d 矩阵

A(:,:,1) =[1     1;  1    1];
A(:,:,2) =[2     2; 2     2];
A(:,:,3) =[4     4; 4     4];

和大小为 2x2 的矩阵 B

B = [ 1 2; 2 3];

我需要的是从 A 中的每个第三维中进行选择使用矩阵 B 的一个数字：

for i=1:2,
  for j=1:2,
    C(i,j) = A(i,j,B(i,j));
  end
end

如何在没有循环的情况下在一行中实现？

Answer 1

不是真正的单行，但没有循环：

[I J] = ind2sub (size(B), 1:numel(B));
linInd = sub2ind (size (A), I, J, B(:)');
C = reshape (A(linInd), size(B));

Answer 2

这是另一种变体：

[r,c,~] = size(A);
[J,I] = meshgrid(1:size(B,1), 1:size(B,2));
idx = reshape(I(:) + r*(J(:)-1) + r*c*(B(:)-1), size(B));
C = A(idx)