从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

您可以使用linear indexing 执行此操作，如下所示：

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