matlab - 从矩阵中提取 block 对角线

Question

我有一个由 nxn 矩阵组成的 njxnj 矩阵。我想提取 nxn 矩阵的对角 j block 。即我想提取对角线(对于 n = 2，j = 4):

最有效的方法是什么？

Answer 1

要为元素建立索引，您可以使用 blkdiag 创建相应的掩码。

%your parameters
n=2
j=4
%some example matrix
M=magic(n*j);
%create the input for blkdiag, j matrices of size n
h=repmat({true(n)},j,1)
%use blkdiag to select the elements
M(logical(blkdiag(h{:})))