时间戳

首页 > 解决方案 > 过滤数据框以保留行，直到列值更改为 0

r - 过滤数据框以保留行，直到列值更改为 0

问题描述

我有这个数据框，

`df <- tibble(
  key = rep(1:3, c(8,4,5)),
  Date = c(seq(as.Date('2016-12-17'), as.Date('2016-12-19'), '1 day'),
       seq(as.Date('2016-12-21'), as.Date('2016-12-23'), '1 day'),
       seq(as.Date('2016-12-29'), as.Date('2016-12-30'), '1 day'),
       seq(as.Date('2017-05-18'), as.Date('2017-05-21'), '1 day'),
       seq(as.Date('2017-05-18'), as.Date('2017-05-20'), '1 day'),
       seq(as.Date('2017-05-23'), as.Date('2017-05-24'), '1 day')),
  flag= c(1,1,1,0,1,1,0,1,1,1,1,1,1,1,1,0,1)
  )`

对于每个“键”，我需要在“标志”第一次变为 0 之前的“日期”的最大值，或者如果“键”没有 0 标志，则返回“日期”的最大值。

我需要以下输出。

key Date        flag
 1 2016-12-19     1
 2 2017-05-21     1
 3 2017-05-20     1

有人可以帮我吗？

谢谢你。

标签： rdplyrdtdate-arithmetic

解决方案

在下面的代码中，我使用了 match 返回第一个匹配项的位置这一事实。然后我用行的索引创建一个额外的列，以便我可以在第一次出现后过滤所有行flag = 1

df %>% 
  group_by(key) %>% 
  arrange(Date) %>% 
  mutate(idx = 1:n()) %>% 
  filter(idx < match(0,flag) | is.na(match(0,flag))) %>% 
  summarise(Date = max(Date))

希望这可以帮助！！

推荐阅读