R中使用IF语句按条件为日期列添加天数报错解决方法

报错原因 #

• 赋值两侧长度不匹配：原代码左侧仅筛选med == "A"的行作为赋值对象，右侧却对整个enddate列全量数据做加14天运算，两侧数据长度不是整数倍关系，因此触发长度不匹配的警告。
• 数据类型不符合运算要求：当前数据框里的startdate、enddate是字符串类型，不是R标准的Date日期类型，直接做数值加天的运算本身逻辑不成立，即使长度匹配也会输出错误结果。

可直接运行的正确代码 #

基础R原生实现（无需安装额外包） #

# 先将字符串格式的日期列转换为标准Date类型
data$startdate <- as.Date(data$startdate)
data$enddate <- as.Date(data$enddate)

# 按条件赋值，保证赋值左右两边筛选的行完全一致，长度匹配
data$enddate[data$med == "A"] <- data$enddate[data$med == "A"] + 14
data$enddate[data$med == "B"] <- data$enddate[data$med == "B"] + 21
# med取值为C的行不做修改，保留原日期

dplyr包实现（逻辑更清晰，不易写错） #

如果习惯用tidyverse语法，可以用case_when写条件分支，从写法上避免长度匹配错误：

library(dplyr)

data <- data %>%
  mutate(
    # 批量转换日期列类型
    across(c(startdate, enddate), as.Date),
    # 按med取值匹配加天规则
    enddate = case_when(
      med == "A" ~ enddate + 14,
      med == "B" ~ enddate + 21,
      TRUE ~ enddate # 其余所有情况保留原enddate值
    )
  )

运行后得到的结果符合预期：med为A的2条记录enddate加14天，med为B的2条记录enddate加21天，med为C的记录保持原值，不会再出现警告。

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