选择包含特定值的行的组（使用 dplyr 和管道）

Question

我正在尝试在分组的 df 中选择包含每个组内特定行上的特定字符串的组。

考虑以下df：

df <- data.frame(id = c(rep("id_1", 4),
                        rep("id_2", 4),
                        rep("id_3", 4)),
                 string = c("here",
                            "is", 
                            "some",
                            "text",
                            "here",
                            "is",
                            "other",
                            "text",
                            "there",
                            "are",
                            "final",
                            "texts"))

我想创建一个数据框，其中包含仅在第二行包含“is”一词的组。

这里有一些不正确的代码：

desired_df <- df %>% group_by(id) %>% 
        filter(slice(select(., string), 2) %in% "is")

这是所需的输出：

desired_df <- data.frame(id = c(rep("id_1", 4),
                                      rep("id_2", 4)),
                               string = c("here",
                                          "is", 
                                          "some",
                                          "text",
                                          "here",
                                          "is",
                                          "other",
                                          "text"))

我查看了here，但这并不能解决我的问题，因为它会找到 any 出现指定字符串的组。

我还可以做一些单独的代码来识别 id，然后使用它来子集原始 df，如下所示：

ids <- df %>% group_by(id) %>% slice(2) %>% filter(string %in% "is") %>% select(id)
desired_df <- df %>% filter(id %in% ids$id)

但我想知道是否可以在单个管道系列中做一些更简单的事情。

帮助表示赞赏！

Answer 1

在按“id”分组后，将第二个元素的“字符串”作为子集，并在 %in% 的 lhs 上应用 %in% 和“is”，以每组返回一个 TRUE

library(dplyr)
df %>%
    group_by(id) %>% 
    filter('is' %in% string[2]) %>%
    ungroup

-输出

# A tibble: 8 x 2
#  id    string
#  <chr> <chr> 
#1 id_1  here  
#2 id_1  is    
#3 id_1  some  
#4 id_1  text  
#5 id_2  here  
#6 id_2  is    
#7 id_2  other 
#8 id_2  text