R dplyr将行保留到列

Question

我有一个像这样的数据框：

> df
  id type
1  1    a
2  1    a
3  1    b
4  1    a
5  1    b
6  2    a
7  2    a
8  2    b
9  3    a
10 3    a

我希望将每个组（id）的所有行保留到第一次出现的值＆＃39; b＆＃39;在类型列中。对于没有类型＆＃39; b＆＃39;的群组，我希望保留所有行。

结果数据框应如下所示：

> dfnew
  id type
1  1    a
2  1    a
3  1    b
4  2    a
5  2    a
6  2    b
7  3    a
8  3    a

我尝试了以下代码，但它保留了其他具有值＆＃39; a＆＃39;超出第一次出现的＆＃39;＆＃39;，并且只排除＆b;＃b;＃的额外事件，这不是我想要的。请看下面的第4行。我想摆脱它。

> df %>% group_by(id) %>% filter(cumsum(type == 'b') <= 1)
Source: local data frame [7 x 2]
Groups: id

  id type
1  1    a
2  1    a
3  1    b
4  1    a
5  2    a
6  2    a
7  2    b
8  3    a
9  3    a

Answer 1

您可以将match或which与slice或（如@Richard所述）which.max

合并

library(dplyr)
df %>% 
  group_by(id) %>% 
  slice(if(any(type == "b")) 1:which.max(type == "b") else row_number())    
# Source: local data table [8 x 2]
# Groups: id
# 
#   id type
# 1  1    a
# 2  1    a
# 3  1    b
# 4  2    a
# 5  2    a
# 6  2    b
# 7  3    a
# 8  3    a

或者您可以使用data.table

进行尝试

library(data.table)
setDT(df)[, if(any(type == "b")) .SD[1:which.max(type == "b")] else .SD, by = id]
#    id type
# 1:  1    a
# 2:  1    a
# 3:  1    b
# 4:  2    a
# 5:  2    a
# 6:  2    b
# 7:  3    a
# 8:  3    a