R语言如何迭代将正定对称矩阵的对称元素对替换为NA

完整实现代码 #

1. 前置依赖与矩阵生成 #

# 加载依赖包（genPositiveDefMat函数来自该包）
library(clusterGeneration)

set.seed(123)
m <- genPositiveDefMat(
    dim = 3, 
    covMethod = "unifcorrmat",
    rangeVar = c(0,1)
)
x <- as.matrix(m$Sigma)
diag(x) <- 1
# 原始矩阵输出
x
# >           [,1]       [,2]       [,3]
# > [1,]  1.0000000 -0.2432303 -0.4110525
# > [2,] -0.2432303  1.0000000 -0.1046602
# > [3,] -0.4110525 -0.1046602  1.0000000

2. 逐轮迭代修改代码 #

通过遍历上三角索引避免修改对角线，同时保证每轮仅修改一对对称非对角元素：

# 提取所有非对角对称对的索引（上三角天然满足r<c，不会包含对角线）
pair_list <- which(upper.tri(x), arr.ind = TRUE)

# 逐轮迭代
for (i in seq(nrow(pair_list))) {
  r <- pair_list[i, 1]
  c <- pair_list[i, 2]
  # 对称位置同时置NA
  x[r, c] <- x[c, r] <- NA
  
  # 可在下方添加每轮迭代后的自定义处理逻辑，此处为打印验证
  cat("第", i, "轮迭代结果：\n")
  print(x)
  cat("\n")
}

运行输出示例 #

第 1 轮迭代结果：
          [,1]       [,2]       [,3]
[1,] 1.0000000         NA -0.4110525
[2,]        NA  1.0000000 -0.1046602
[3,] -0.4110525 -0.1046602  1.0000000

第 2 轮迭代结果：
          [,1]       [,2]       [,3]
[1,] 1.0000000         NA         NA
[2,]        NA  1.0000000 -0.1046602
[3,]        NA -0.1046602  1.0000000

第 3 轮迭代结果：
     [,1] [,2] [,3]
[1,]    1   NA   NA
[2,]   NA    1   NA
[3,]   NA   NA    1

原有代码问题说明 #

1. 未排除r==c的对角线场景，导致对角线被错误修改，本方案仅遍历上三角元素，天然避开对角线
2. 原有逻辑将判断语句和赋值语句混写，循环逻辑错误，本方案提前生成所有待修改索引列表，循环逻辑清晰无异常退出问题

快捷批量处理方案 #

如果不需要逐轮迭代，直接一次性将所有非对角元素置为NA，可直接用一行代码实现：

x[upper.tri(x) | lower.tri(x)] <- NA

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