table()返回行数不一致：如何生成含顺序补0的计数数据框？

问题背景 #

你需要按概率c(.3,.4,.3)对数值0、1、2进行10次抽样（每次抽10个样本），统计每个抽样批次中三个数值的出现次数，生成结构统一的数据框。但当前使用的代码：

replicate(10,table(sample(x=c(0,1,2), size=10, replace=TRUE, prob=c(.3,.4,.3))))

会因部分批次只抽到2个数值，导致转换为dataframe时因行列数不匹配报错。

解决方案 #

方法1：抽样时强制保留所有目标水平（推荐） #

通过factor将抽样结果转换为包含0、1、2所有水平的因子，这样table会自动为未抽到的数值统计0次，避免列缺失：

# 定义目标数值的水平（确保顺序和你需要的一致）
target_vals <- c("0", "1", "2")

# 执行抽样并计数，强制保留所有水平
count_matrix <- replicate(10, 
                          table(factor(sample(x=c(0,1,2), size=10, replace=TRUE, prob=c(.3,.4,.3)),
                                       levels = target_vals)))

# 转置矩阵并转换为数据框（每一行对应一个抽样批次的计数）
result_df <- as.data.frame(t(count_matrix))

方法2：对已有结果手动补全0 #

如果已经得到了原始的replicate结果（需设置simplify=FALSE返回列表），可以遍历每个批次的计数，补全缺失数值的0：

# 获取原始抽样计数结果（返回列表，避免长度不匹配报错）
raw_counts <- replicate(10, 
                        table(sample(x=c(0,1,2), size=10, replace=TRUE, prob=c(.3,.4,.3))),
                        simplify = FALSE)

# 遍历每个批次，补全0、1、2的计数
filled_counts <- lapply(raw_counts, function(batch) {
  # 创建初始全0的计数向量，命名为目标数值
  full_count <- setNames(numeric(3), c("0", "1", "2"))
  # 将已有计数赋值到对应位置
  full_count[names(batch)] <- batch
  full_count
})

# 合并为数据框
result_df <- as.data.frame(do.call(rbind, filled_counts))

两种方法最终都会生成结构一致的数据框，每一行对应一个抽样批次，列分别为0、1、2的出现次数。

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