如何为数据框特定观测行创建虚拟变量（伍德里奇计量习题场景）

问题背景 #

正在完成伍德里奇《计量经济学导论：现代方法》第9章C12(iii)习题，需求为：

1. 筛选出变量bs大于0.5的所有观测值
2. 为这些观测值分别创建对应的回归用虚拟变量

已完成第一步，通过以下代码定位到目标观测行：

library('wooldridge')
which(elem94_95$bs>0.5)
# [1]   68 1127 1508 1670

但尝试创建虚拟变量时代码无法运行：

elem94_95$d68<-ifelse(row==68,1,0)

解决方法 #

报错原因是R中没有默认的row变量，需要明确指定数据框的行号序列。以下两种方法可实现需求：

方法1：手动逐个创建 #

针对每个目标行号，生成对应虚拟变量：

# 获取数据框的完整行号序列
row_nums <- seq_len(nrow(elem94_95))

# 为第68行创建虚拟变量d68
elem94_95$d68 <- ifelse(row_nums == 68, 1, 0)
# 为第1127行创建虚拟变量d1127
elem94_95$d1127 <- ifelse(row_nums == 1127, 1, 0)
# 为第1508行创建虚拟变量d1508
elem94_95$d1508 <- ifelse(row_nums == 1508, 1, 0)
# 为第1670行创建虚拟变量d1670
elem94_95$d1670 <- ifelse(row_nums == 1670, 1, 0)

方法2：批量创建（高效版） #

如果目标行号较多，用循环批量生成更便捷：

target_rows <- c(68, 1127, 1508, 1670)
row_nums <- seq_len(nrow(elem94_95))

for (row in target_rows) {
  # 生成虚拟变量名，比如d68
  var_name <- paste0("d", row)
  # 为对应行赋值1，其余行赋值0
  elem94_95[[var_name]] <- ifelse(row_nums == row, 1, 0)
}

验证结果 #

可通过以下代码检查虚拟变量是否正确生成：

# 查看目标行的虚拟变量取值
elem94_95[target_rows, c("d68", "d1127", "d1508", "d1670")]

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