在Matlab中從3D矩陣和索引數組中提取2D矩陣

Question

我找不到滿意的答案。 如果A是大小為（m，n，k）的3D矩陣，Z是大小為mxn（值在1和k之間的整數）的2D矩陣，我想提取定義如下的S：

for i=1:m
    for j=n
        S(i,j) = A(i,j,Z(i,j));
    end
end

是否有一種有效的（矢量化）方式來做到這一點？

先感謝您

Answer 1

您可以使用線性索引進行如下操作：

S = reshape(A((1:m*n).' + m*n*(Z(:)-1)), m, n);