快速操作R

时间:2019-03-14 04:53:01

标签: r lubridate

我有大约34000个日期向量,必须更改日期并移动月份。我已经尝试过使用循环并使用mapply函数,但这非常慢。 这是我所拥有的一个例子:

library(lubridate)
list_dates = replicate(34000,seq(as.Date("2019-03-14"),length.out = 208,by = "months"),simplify = F)
new_day = round(runif(34000,1,30))
new_day[sample(1:34000,10000)] = NA

new_dates = mapply(FUN = function(dates,day_change){
  day(dates) = ifelse(is.na(rep(day_change,length(dates))),day(dates),rep(day_change,length(dates)))
  dates = as.Date(ifelse(is.na(rep(day_change,length(dates))),dates,dates%m-%months(1)),origin = "1970-01-01")
  return(dates)
},dates = list_dates,day_change = as.list(new_day),SIMPLIFY = F)

变量new_dates应该包含一个原始日期列表,该日期相应地移动到了变量new_day。侧面的功能如下:

  1. 如果new_day与NA不同,它将日期更改为新日期。
  2. 如果new_day与NA不同,它将把日期的月份推迟一个月。

无论采用哪种软件包(如果它们在CRAN中),我都会接受任何可以提高速度的解决方案。

编辑

因此,根据注释,我将示例简化为一个包含2个日期向量的列表,每个日期向量包含2个日期,并创建了一个手动的新日期向量:

list_dates = replicate(2,seq(as.Date("2019-03-14"),length.out = 2,by = "months"),simplify = F)

new_day = c(9,NA)

这是原始输入(变量list_dates):

[[1]]
[1] "2019-03-14" "2019-04-14"

[[2]]
[1] "2019-03-14" "2019-04-14"

mapply函数的预期输出为:

[[1]]
[1] "2019-02-09" "2019-03-09"

[[2]]
[1] "2019-03-14" "2019-04-14"

您可以看到,第一个日期向量更改为第9天,每个日期都滞后一个月。日期的第二个向量没有改变,因为该值的new_datesNA

2 个答案:

答案 0 :(得分:0)

这是一个lubridate解决方案

library(lubridate)
mapply(
    function(x, y) { if (!is.na(y)) {
            day(x) <- y;
            month(x) <- month(x) - 1
        }
        return(x) },
    list_dates, new_day, SIMPLIFY = F)
#[[1]]
#[1] "2019-02-09" "2019-03-09"
#
#[[2]]
#[1] "2019-03-14" "2019-04-14"

或使用purrr

library(purrr)
library(lubridate)
map2(list_dates, new_day, function(x, y) {
    if (!is.na(y)) {
        day(x) <- y
        month(x) <- month(x) - 1
    }
    x })

答案 1 :(得分:0)

除了Maurits的解决方案之外,如果您想进一步提高计算速度,则可能需要考虑将多个核与doParallel一起使用

library(data.table)
library(doParallel)

registerDoParallel(3)
df <- data.table(new_day,list_dates)

mlply(df,
      function(new_day,list_dates){

        list_dates <- list_dates[[1]]

        if(!is.na(new_day)){
          day(list_dates) <- new_day
          list_dates <-  list_dates %m-% months(1)
        }

        return(list_dates)
      }, .parallel = T, .paropts = list(.packages='lubridate')
)