R语言如何将数据框choice列的0值替换为可选值集合内的有效值

首先修正可选项目数据集的构造问题：你提供的choices构造代码seq(600:800)存在语法问题，会生成重复值，正确的构造方式如下：

# 生成600到800的连续整数作为可选项目值
choices <- data.frame(choices = seq(600, 800))
# 提取可选值为向量方便后续调用
valid_choices <- choices$choices

方案1：基础R实现 #

如果不需要保证同一个学生的两个志愿不重复，直接批量替换0值即可：

# 替换choice_1列的0值
df$choice_1[df$choice_1 == 0] <- sample(valid_choices, size = sum(df$choice_1 == 0), replace = TRUE)
# 替换choice_2列的0值
df$choice_2[df$choice_2 == 0] <- sample(valid_choices, size = sum(df$choice_2 == 0), replace = TRUE)

如果需要保证同一学生的两个志愿不同，用循环处理：

for (i in 1:nrow(df)) {
  # 仅处理第二志愿为0的行
  if (df$choice_2[i] == 0) {
    # 排除当前学生的第一志愿后再抽样
    available_choices <- valid_choices[valid_choices != df$choice_1[i]]
    df$choice_2[i] <- sample(available_choices, size = 1)
  }
  # 若第一志愿也可能存在0值，可在循环内补充对应判断逻辑
}

方案2：dplyr包实现（代码更简洁） #

library(dplyr)

df <- df %>%
  # 按行处理保证每个学生的志愿独立抽样
  rowwise() %>%
  mutate(
    choice_1 = ifelse(choice_1 == 0, sample(valid_choices, 1), choice_1),
    # 第二志愿抽样时自动排除当前学生的第一志愿
    choice_2 = ifelse(choice_2 == 0, sample(valid_choices[valid_choices != choice_1], 1), choice_2)
  ) %>%
  ungroup()

效果示例 #

你提供的示例数据处理后输出类似如下（抽样值随机，结果不唯一）：

如果需要所有学生的志愿全局不重复，可将sample的replace参数设置为FALSE，需保证可选项目的总数大于需要替换的0值总数量。

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