R语言如何使用while循环创建符合分组规则的虚拟变量

首先修正示例输入中的向量笔误：原代码X1<-c(1,1,2,2,,3,3)存在多余空逗号，会导致向量长度不匹配，修正后的基础输入为：

# 基础输入
X1 <- c(1, 1, 2, 2, 3, 3)
X2 <- c(0, 1, 0, 0, 1, 1)
# 初始化X3为与输入等长的全0向量
X3 <- rep(0, length(X1))

实现逻辑 #

按X1的独立分类逐个遍历，对每个分类执行判断：

• 若该分类下所有观测中存在至少1行X2取值为1，则该分类下所有行的X3设为1
• 若该分类下所有观测的X2均为0，则X3保持为初始的0值

对应while循环的实现代码：

# 提取X1的所有不重复分类值
unique_cats <- unique(X1)
# 初始化循环计数索引
loop_idx <- 1
# while循环主体
while(loop_idx <= length(unique_cats)){
  # 当前遍历的分类值
  cur_category <- unique_cats[loop_idx]
  # 定位该分类对应的所有观测行位置
  target_rows <- which(X1 == cur_category)
  # 检查该分类下是否存在X2=1的观测
  if(any(X2[target_rows] == 1)){
    X3[target_rows] <- 1
  }
  # 计数自增，避免触发无限循环
  loop_idx <- loop_idx + 1
}

结果校验 #

拼接三个变量生成数据框，运行输出如下，与预期结果完全匹配：

# 拼接输出结果
result <- data.frame(X1 = X1, X2 = X2, X3 = X3)
print(result)

输出内容：

X1 X2 X3
1  1  0  1
2  1  1  1
3  2  0  0
4  2  0  0
5  3  1  1
6  3  1  1

操作注意点 #

• 循环内必须对计数索引做自增操作，否则会进入无限循环导致程序无响应
• 初始化X3时需保证向量长度与X1、X2完全一致，否则会触发向量长度不匹配的报错
• 使用any()判断分类下是否存在X2=1的记录，比逐行遍历判断的执行效率更高

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