如何在R语言中根据列名创建period列？解决赋值行数不匹配错误

问题背景 #

给定如下样本数据：

stfips <- c("39","39","39")
year <- c("2023", "2023","2023")
industry_code <- c(112, 113, 114)
first_quarter_establishments <- c(987,654,321)
county <- data.frame(stfips, year, industry_code, first_quarter_establishments)

需求为创建period列：若数据框存在含“first”的列名则值为"01"，含“second”则为"02"，以此类推。使用ChatGPT提供的代码时出现如下错误：

Error in $<-.data.frame(*tmp*, period, value = c("", "", "", "01")) : replacement has 4 rows, data has 3

错误原因 #

原代码中grepl("first", names(county), ignore.case = TRUE)返回的是与数据框列数等长的逻辑向量（此处为4个元素），而数据框仅有3行。直接用ifelse生成的向量长度与数据框行数不匹配，导致赋值失败。

解决方法 #

方法1：单季度场景（仅判断是否含"first"列） #

先检查数据框是否存在符合条件的列，再为整个period列统一赋值：

# 检查是否存在含"first"的列
has_first_quarter <- any(grepl("first", names(county), ignore.case = TRUE))
# 生成period列
county$period <- if(has_first_quarter) "01" else ""

方法2：多季度通用场景 #

如果数据框可能包含不同季度的列（如含"second"/"third"/"fourth"的列），可通过映射关系实现通用处理：

# 定义季度关键词与period的映射表
quarter_map <- c(
  "first" = "01",
  "second" = "02",
  "third" = "03",
  "fourth" = "04"
)

# 匹配数据框中存在的季度关键词
matched_quarters <- names(quarter_map)[grepl(
  paste(names(quarter_map), collapse = "|"), 
  names(county), 
  ignore.case = TRUE
)]

# 生成period列（若存在多个匹配，默认取第一个；可按需调整逻辑）
county$period <- if(length(matched_quarters) > 0) quarter_map[matched_quarters[1]] else ""

验证结果 #

执行上述代码后，数据框将生成符合预期的period列：

stfips year industry_code first_quarter_establishments period
1     39 2023           112                          987     01
2     39 2023           113                          654     01
3     39 2023           114                          321     01

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