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

Question

我有一个 M×M×N 矩阵，它是 N 个 M×M 矩阵的串联。我想通过取每个 M×M 子矩阵的对角线并将它们连接在一起，将该矩阵简化为 M×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 及更高版本时才有效。如果您的版本比该版本旧，请改用虚拟变量。