R语言实现矩阵列非NA值上移、NA值移至列底部的方法

现有如下矩阵：

c1   c2   c3   c4   c5
r1  1    NA   NA   NA   NA
r2  1    1    NA   NA   NA
r3  1    1    1    NA   NA
r4  1    1    1    1    NA
r5  1    1    1    1    1

需求为：将每列中的NA值移至列底部，同时将非NA值上移，且不改变非NA值的原有顺序（假设非NA值均唯一），处理后矩阵如下：

c1   c2   c3   c4   c5
r1  1    1    1    1    1
r2  1    1    1    1    NA
r3  1    1    1    NA   NA
r4  1    1    NA   NA   NA
r5  1    NA   NA   NA   NA

疑问：

• 是否存在可实现该需求的R语言函数？此前找到类似问题但需求相反，无法复用其解决方案。
• 能否通过循环或将元素转存至新矩阵的方式实现该需求？

解决方案 #

1. 基础R快速实现（无额外依赖） #

R没有专门对应此需求的内置函数，但用apply()可以简洁实现：

# 构造示例矩阵
mat <- matrix(
  c(1,1,1,1,1,
    NA,1,1,1,1,
    NA,NA,1,1,1,
    NA,NA,NA,1,1,
    NA,NA,NA,NA,1),
  nrow = 5, ncol = 5,
  dimnames = list(paste0("r",1:5), paste0("c",1:5))
)

# 按列处理：提取非NA值，后续补全NA，最后转置恢复原矩阵结构
processed_mat <- t(apply(mat, 2, function(col) {
  c(col[!is.na(col)], rep(NA, length(col) - sum(!is.na(col))))
}))

2. 循环实现 #

如果偏好直观的流程控制，用for循环逐列处理也可行：

processed_mat_loop <- mat
for (col_idx in 1:ncol(processed_mat_loop)) {
  current_col <- processed_mat_loop[, col_idx]
  non_na_vals <- current_col[!is.na(current_col)]
  processed_mat_loop[, col_idx] <- c(non_na_vals, rep(NA, nrow(processed_mat_loop) - length(non_na_vals)))
}

3. 转存新矩阵实现 #

若需要保留原始矩阵不被修改，可直接创建新矩阵填充数据：

new_mat <- matrix(NA, nrow = nrow(mat), ncol = ncol(mat), dimnames = dimnames(mat))
for (col_idx in 1:ncol(mat)) {
  non_na_vals <- mat[, col_idx][!is.na(mat[, col_idx])]
  new_mat[1:length(non_na_vals), col_idx] <- non_na_vals
}

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