改变每组匹配子集条件的行数

Question

我希望改变一个名为 SF_COUNT 的新列，它是列类型包含“SF”的每组行数的每组计数 (ID)

一个可重现的例子如下：

df <- data.frame(ID = c(1234,1234,1234,4567,4567,4567,4567,8900,8900,8900),type = c('RF','SF','SF','RF','SF','SF','SF','RF','SF','SF'))

我的最终数据框如下所示：

final_df <- data.frame(ID = c(1234,1234,1234,4567,4567,4567,4567,8900,8900,8900),type = c('RF','SF','SF','RF','SF','SF','SF','RF','SF','SF'), SF_COUNT = c(2,2,2,3,3,3,3,2,2,2))

请问如何在 dplyr 中实现这一点？

Answer 1

按'ID'分组后，获取mutate中逻辑vector（type == 'SF'）的sum创建新列

library(dplyr)
df <- df %>% 
    group_by(ID) %>% 
    mutate(SF_COUNT = sum(type == 'SF', na.rm = TRUE))

---

如果是子串，则使用str_detect

library(stringr)
df <- df %>% 
    group_by(ID) %>% 
    mutate(SF_COUNT = sum(str_detect(type, 'SF'), na.rm = TRUE))