將日期聚合為 R 中的日期間隔/周期

Question

我有以下示例數據：

require(tibble)
sample_data <- tibble(
                      emp_name = c("john", "john", "john", "john","john","john", "john"), 
                      task = c("carpenter", "carpenter","carpenter", "painter", "painter", "carpenter", "carpenter"),
                      date_stamp = c("2019-01-01","2019-01-02", "2019-01-03", "2019-01-07", "2019-01-08", "2019-01-30", "2019-02-02")
                      )

為此，我需要根據日期聚合成間隔。

規則是：如果為同一屬性列出的下一個date_stamp之間沒有日期，那么它應該被聚合。 否則， date_stamp_from和date_stamp_to應該等於date_stamp 。

desired_result <- tibble(
                  emp_name = c("john", "john","john", "john"),
                  task = c("carpenter","painter", "carpenter", "carpenter"),
                  date_stamp_from = c("2019-01-01","2019-01-07", "2019-01-30", "2019-02-02"),
                  date_stamp_to = c("2019-01-03","2019-01-08", "2019-01-30", "2019-02-02"),
                  count_dates = c(3,2,1,1)
)

解決這個問題的最有效方法是什么？ 原始數據集大約有 10000 條記錄。

Answer 1

我們可以使用diff和cumsum創建組並計算每個組中的first 、 last date_stamp和行數。

library(dplyr)

sample_data %>%
     mutate(date_stamp = as.Date(date_stamp)) %>%
     group_by(gr = cumsum(c(TRUE, diff(date_stamp) > 1))) %>%
     mutate(date_stamp_from = first(date_stamp), 
            date_stamp_to = last(date_stamp), 
            count_dates = n()) %>%
     slice(1L) %>%
     ungroup() %>%
     select(-gr, -date_stamp)

# A tibble: 4 x 5
#  emp_name task      date_stamp_from date_stamp_to count_dates
#  <chr>    <chr>     <date>          <date>              <int>
#1 john     carpenter 2019-01-01      2019-01-03              3
#2 john     painter   2019-01-07      2019-01-08              2
#3 john     carpenter 2019-01-30      2019-01-30              1
#4 john     carpenter 2019-02-02      2019-02-02              1