更改屬於同一組的行的值並且該組中至少有一個觀察符合 dplyr 的條件

Question

假設我有這個 dataframe：

df <- data.frame(Sequence_ID = c(100,100,100,100,101,101,101,101,102,102,102,102,103,103,103,103), Success = c(1,0,0,0,0,0,0,0,0,0,1,0,0,0,0,1))

如果任何具有相同 Sequence_ID 的行在 Success 列中有 1，那么我希望該組中的所有行在成功列中有 1。

我可以使用以下代碼獲得我想要的 output：

for(i in 1:nrow(df)){
  x <- df$Sequence_ID[i]
  if (any(df$success[df$Sequence_ID == x] == 1)){
    df$success[df$Sequence_ID == x] <- 1
  }
}

我想知道在 dplyr 中是否有辦法做到這一點。 提前致謝

Answer 1

library(dplyr)

df %>%
  group_by(Sequence_ID) %>%
  mutate(
    Success = as.integer(any(as.logical(Success)))
  )

訣竅:)