-2

从这篇文章扩展这个问题

两个日期之间的差异,不包括周末和 R 中给定的假期列表

如果我有特定于州的假期怎么办。我如何按州合并假期?

holiday <- data.frame(h = as.Date(c("2016/05/3", "2016/05/3"),'%Y/%m/%d'), 
                      s = c('state1','state2'),
                      stringsAsFactors=FALSE
                      )

d <- data.frame(d1 = as.Date(c('2016/05/2','2016/05/2'),'%Y/%m/%d'),
                d2= as.Date(c('2016/05/10','2016/05/10'),'%Y/%m/%d'),
                s = c('state1','state3'),
                stringsAsFactors=FALSE)

生产

> d
          d1         d2      s
1 2016-05-02 2016-05-10 state1
2 2016-05-02 2016-05-10 state3
> holiday
           h      s
1 2016-05-03 state1
2 2016-05-03 state2

链接中提供的解决方案是特定于日期的假期,但没有考虑到各州。我怎样才能合并国家。

Desired state 是 d 中的一个额外列,用于计算不包括特定于州的周末和节假日的天数

4

1 回答 1

1

这是我的问题的一个不雅的解决方案。希望您觉得这个有帮助。

library('dplyr')

holiday <- data.frame(h = as.Date(c("2016/05/3", "2016/05/3"),'%Y/%m/%d'), 
                      s = c('state1','state2'),
                      stringsAsFactors=FALSE
                      )

d <- data.frame(d1 = as.Date(c('2016/05/2','2016/05/2'),'%Y/%m/%d'),
                d2= as.Date(c('2016/05/10','2016/05/10'),'%Y/%m/%d'),
                s = c('state1','state3'),
                stringsAsFactors=FALSE)

state <- as.character(unique(d$s))

#function to exclude weekends and holidays
f <- function(a, b, h) { 
  d <- seq(a, b, 1)[-1]   
  sum(!format(d, "%u") %in% c("6", "7") & !d %in% h)
}
vf <- Vectorize(f, c("a", "b"))

#Function to calculate days excluding weekends and holidays by state
#state is done iteratively
datalist = list()
for (i in 1:length(state)) {
  # ... make some data

  #extract holidays by state as vectors
  holi <- holiday %>% 
    filter(s== state[i]) %>%
    select(h) %>%
    pull()

  #create new data by state
  dat <- d %>%
    filter(s == state[i])
  dat$diff <- with(dat,vf(d1, d2, holi))

  datalist[[i]] <- dat # add it to your list
  rm('dat')
}

new_df = dplyr::bind_rows(datalist)

通过特定的州区分假期并排除周末

          d1         d2      s diff
1 2016-05-02 2016-05-10 state1    5
2 2016-05-02 2016-05-10 state3    6
于 2018-03-28T11:58:06.553 回答