R语言：分组DataFrame中按条件替换首行字符串的实现求助

问题描述 #

现有按id变量分组的DataFrame，需求如下：

• 对每个id对应的X、Y、Z列
• 若该id组内除首行外的其他行存在"yes"，则将首行的"no"替换为"yes"

示例数据 #

id <- c(1,1,1,2,2,3,3)
X <- c("yes", "no", "no", "no", "no", "no", "no")
Y <- c("no", "no", "yes", "no", "yes", "no", "no")
Z <- c("no", "yes", "no", "no", "no", "no", "no")
df <- data.frame(id, X, Y, Z)

期望输出 #

id   X   Y   Z
 1 yes yes yes
 1  no  no  no
 1  no  no  no
 2  no yes  no
 2  no  no  no
 3  no  no  no
 3  no  no  no

解决方案 #

方法1：使用dplyr包（简洁高效） #

借助分组操作和across函数批量处理多列：

library(dplyr)

df_modified <- df %>%
  group_by(id) %>%
  mutate(across(c(X, Y, Z), ~ {
    # 检查组内除首行外是否存在"yes"
    has_yes <- any(.x[-1] == "yes")
    # 首行满足条件则替换，否则保持原内容
    if(row_number() == 1 && .x == "no" && has_yes) "yes" else .x
  })) %>%
  ungroup()

print(df_modified)

代码说明

• group_by(id)：按id拆分数据组
• across(c(X, Y, Z), ~ {...})：对指定列统一应用处理逻辑
• .x[-1]：取当前列除首行外的所有行
• row_number() == 1：定位组内首行，结合条件完成替换

方法2：基础R实现（无需额外包） #

通过ave函数结合循环处理列：

df_modified_base <- df
target_cols <- c("X", "Y", "Z")

for(col in target_cols) {
  df_modified_base[[col]] <- ave(df_modified_base[[col]], df_modified_base$id, FUN = function(x) {
    has_yes <- any(x[-1] == "yes")
    if(has_yes && x[1] == "no") {
      x[1] <- "yes"
    }
    x
  })
}

print(df_modified_base)

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