如何将现有协方差矩阵按指定分组转换为分块对角矩阵？

现有一个协方差矩阵，希望根据各列所属的分组（比如示例中前2行/列属于组1，后续属于组2）将其转换为分块对角矩阵，有没有简便的实现方法？

示例数据：

m1 <- matrix(1:16, ncol=4, byrow=TRUE)
rownames(m1) <- colnames(m1 ) <- c('a', 'b', 'c', 'd')

原矩阵输出：

a  b  c  d
a  1  2  3  4
b  5  6  7  8
c  9 10 11 12
d 13 14 15 16

分组情况：

• Group 1: 'a', 'b'
• Group 2: 'c', 'd'

期望结果：

a  b  c  d
a  1  2  0  0
b  5  6  0  0
c  0  0 11 12
d  0  0 15 16

解决方案 #

方法1：掩码矩阵法（通用分组场景） #

先基于分组生成逻辑掩码，保留同一组内的元素，其余置0：

# 定义分组列表
groups <- list(Group1 = c('a', 'b'), Group2 = c('c', 'd'))

# 创建初始全为FALSE的掩码矩阵
mask <- matrix(FALSE, nrow(m1), ncol(m1))

# 为每个分组对应的行列交叉区域标记为TRUE
for (g in groups) {
  idx <- rownames(m1) %in% g
  mask[idx, idx] <- TRUE
}

# 生成分块对角矩阵
block_diag_m1 <- m1 * mask

方法2：直接赋值法（位置明确场景） #

如果已知分组对应的行列位置，直接清除跨组区域的元素：

# 清除组1与组2的交叉区域
m1[1:2, 3:4] <- 0
m1[3:4, 1:2] <- 0

这种方式代码更简洁，适合分组位置固定且数量少的情况。

方法3：Matrix包批量组合法（多分组场景） #

如果分组数量较多，用Matrix包的bdiag函数批量组合子矩阵：

library(Matrix)

# 提取每个分组对应的子矩阵
sub_mats <- lapply(groups, function(g) m1[g, g])

# 组合成分块对角矩阵并转换为普通矩阵
block_diag_m1 <- as.matrix(bdiag(sub_mats))

# 还原原行列名
rownames(block_diag_m1) <- colnames(block_diag_m1) <- unlist(groups)

内容的提问来源于stack exchange，提问作者Tartaglia