如何移除矩阵中的NA值并保持矩阵为方阵格式？

要移除矩阵中所有含NA的行和对应列，保持方阵结构，你可以按以下步骤操作：

1. 识别含NA的行和列
   先找出所有包含NA的行索引和列索引：

   # 找出含NA的行
   na_rows <- which(apply(LD_matrix, 1, anyNA))
   # 找出含NA的列
   na_cols <- which(apply(LD_matrix, 2, anyNA))
   

2. 同时移除行和列
   从原矩阵中剔除这些行和列，得到干净的方阵：

   clean_LD_matrix <- LD_matrix[-na_rows, -na_cols]
   

3. 验证结果
   检查处理后的矩阵维度：

   dim(clean_LD_matrix)
   

   此时应该输出[1] 5695 5695，符合你的预期。

简化优化（针对对称矩阵） #

如果你的LD矩阵是对称矩阵（绝大多数LD矩阵都是对称的），含NA的行和列索引完全一致，你可以用更简洁的代码：

# 合并并去重含NA的行/列索引
na_indices <- unique(c(which(apply(LD_matrix, 1, anyNA)), which(apply(LD_matrix, 2, anyNA))))
# 移除对应行和列
clean_LD_matrix <- LD_matrix[-na_indices, -na_indices]

为什么na.omit不行？ #

na.omit()函数的设计逻辑是仅移除包含NA的行，不会处理列，所以会导致处理后矩阵变成矩形而非方阵。而你的场景需要同时剔除有NA的行和对应列，才能保持方阵的结构。

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