R语言如何按随机比例拆分给定总数生成指定份数的样本数

核心实现逻辑 #

• 先生成和拆分份数等量的非负随机数
• 对随机数做归一化处理：每个数除以所有随机数的总和，自动得到和为1的随机比例
• 用预设总数值乘以每个归一化后的比例，得到的拆分结果总和必然等于原总数值

具体代码实现 #

通用浮点数值拆分函数 #

# 参数说明：
# total_num: 预设的总数值
# split_n: 要拆分的份数
split_random <- function(total_num, split_n) {
  # 1. 生成split_n个非负随机数，默认用runif生成0-1之间的均匀随机数
  rand_nums <- runif(n = split_n, min = 0, max = 1)
  # 如果要用到你熟悉的sample函数，可以替换为下面这行，生成1到100的随机整数：
  # rand_nums <- sample(x = 1:100, size = split_n, replace = TRUE)
  
  # 2. 归一化得到和为1的随机比例
  ratio <- rand_nums / sum(rand_nums)
  
  # 3. 计算拆分后的数值
  split_result <- total_num * ratio
  
  # 返回比例和拆分结果
  return(list(ratio = ratio, split_result = split_result))
}

整数拆分专用函数（贴合示例需求） #

如果需要拆分结果为整数，可补充调整步骤，完全消除浮点误差，保证总和和原数值完全一致：

split_random_integer <- function(total_num, split_n) {
  rand_nums <- runif(split_n)
  ratio <- rand_nums / sum(rand_nums)
  # 对每个拆分值向下取整
  split_result <- floor(total_num * ratio)
  # 计算取整后的总和与原总数值的差值，补到最后一个数上
  diff <- total_num - sum(split_result)
  split_result[split_n] <- split_result[split_n] + diff
  return(split_result)
}

示例运行 #

按照你给出的测试场景，总数值100、拆分5份：

# 调用浮点拆分函数
res <- split_random(total_num = 100, split_n = 5)
# 查看随机比例
res$ratio
# 示例输出（每次运行结果随机）：[1] 0.18 0.07 0.25 0.32 0.18
# 查看拆分结果
res$split_result
# 对应输出：[1] 18 7 25 32 18
# 验证总和
sum(res$split_result)
# 输出：100

# 调用整数拆分函数
int_res <- split_random_integer(100, 5)
int_res
# 示例输出：[1] 15 22 8 35 20
sum(int_res)
# 输出：100

现有零散代码的整合说明 #

• 你提到的sample(x:xx)可以直接用来生成初始随机数，只要保证生成的数为非负，替换代码中runif的位置即可
• 不需要用到seq，因为seq是固定步长生成序列，不符合随机拆分的需求
• 归一化步骤已经自动保证比例和为1，不需要手动调整数值凑总和

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