如何在MATLAB中索引3-D矩阵的对角线？

Question

我有一个M乘M乘N矩阵，它是N个M乘M矩阵的串联。我想通过取每个M-x-M子矩阵的对角线并将它们连接在一起，将这个矩阵简化为M-x-N矩阵。我如何才能以一种简单的矢量化方式做到这一点呢？

Answer 1

您可以通过获取对角线的线性索引并使用它来形成一个新的矩阵来完成此操作

[M,~,N]=size(A);%# A is your matrix
indx=cumsum([1:(M+1):M^2; M^2.*ones(N-1,M)]);%#diagonal indices

B=A(indx');%'# transpose to get MxN

在上面，我使用了~来忽略函数的输出。然而，只有当你使用MATLAB R2009b或更高版本时，这才能起作用。如果您的版本早于此版本，请改用伪变量。