过滤组内的数据帧，其中一列满足 R 中的 AND 条件

Question

我有以下数据框，我只需要过滤那些每组 (id) 都有“摄入量”和“排放量”的行。结果应该是这样的：

> df <- tibble(id = c(1, 1, 2, 3, 3, 4, 4, 5, 6, 7, 7),
+              type = c("intake", "discharge", "intake", "intake", "discharge",
+                       "intake", "discharge", "intake", "intake", "intake", "discharge"))
> df
      id type     
   <dbl> <chr>    
 1     1 intake   
 2     1 discharge
 3     2 intake   
 4     3 intake   
 5     3 discharge    
 6     4 intake   
 7     4 discharge
 8     5 intake   
 9     6 intake   
10     7 intake   
11     7 discharge

为此：

      id type     
   <dbl> <chr>    
 1     1 intake   
 2     1 discharge   
 3     3 intake   
 4     3 discharge    
 5     4 intake   
 6     4 discharge 
 7     7 intake   
 8     7 discharge

因此，没有同时具有摄入量和排放量的组 (id) 将被删除（并且仅保留具有两者的组）。

我希望这是有道理的...抱歉让我度过了漫长的一天。

Answer 1

library(dplyr)
df %>%
  group_by(id) %>%
  filter(sum(type == "intake") >= 1,
         sum(type == "discharge") >= 1) %>%
  # add below if we only want intake/discharge lines
  # filter(type %in% c("intake", "discharge")) %>% 
  ungroup()

结果（因在 OP 中添加“其他”而有所不同，所需行为不明确）

# A tibble: 9 x 2
     id type     
  <dbl> <chr>    
1     1 intake   
2     1 discharge
3     3 intake   
4     3 discharge
5     3 other    
6     4 intake   
7     4 discharge
8     7 intake   
9     7 discharge

Answer 2

这是 select 有 "intake" 和 "discharge" 的组的方法。

library(dplyr)

values <- c('intake', 'discharge')

df %>%
  group_by(id) %>%
  filter(all(values %in% type) & type %in% values) %>%
  ungroup

#     id type     
#  <dbl> <chr>    
#1     1 intake   
#2     1 discharge
#3     3 intake   
#4     3 discharge
#5     4 intake   
#6     4 discharge
#7     7 intake   
#8     7 discharge

all(values %in% type) selects 具有两个值的完整组，而 type %in% values 将 select 在具有两个值之一的那些组行中。

过滤组内的数据帧，其中一列满足 R 中的 AND 条件

Filter dataframe within a group with one column meeting an AND condition in R

r

subset

filter

dplyr