按ID和日期合并两个数据框

时间:2020-08-02 19:57:14

标签: r dplyr data.table

我有两个数据帧。 答:

id  date
1   2010-05-08
2   2012-08-08
3   2013-06-23

B:

id  date1
1   2010-05-09
2   2012-08-08

我需要按ID合并两个数据帧,并且还要将表2中的日期=表1中的日期+1天。进一步标记合并为TRUE的行。

最终输出是 答:

id  date           date1      flag
1   2010-05-08    2010-05-09   1
2   2012-08-08       NA        NA
3   2013-06-23       NA        NA

用于生成数据的代码-

A <- data.frame(customer = c(1,2,3),
                application_date = c("2010-05-08", "2012-08-08", "2013-06-23"))
B <- data.frame(customer = c(1,2),
                application_date = c("2010-05-09", "2012-08-08"))

3 个答案:

答案 0 :(得分:1)

怎么样?

数据

A <- data.frame(customer = c(1,2,3),
                application_date = c("2010-05-08", "2012-08-08", "2013-06-23"))
B <- data.frame(customer = c(1,2),
                application_date = c("2010-05-09", "2012-08-08"))

DPLYR

library(dplyr)

    data <- left_join(A, B, by = "customer")
    data %>% 
      mutate(logic = if_else(as.Date(data$application_date.x) + 1 == as.Date(data$application_date.y), 1, 0)) %>% 
      rename("id" = "customer",
             "date" = "application_date.x",
             "date1" = "application_date.y",
             "flag" = "logic")

输出

 id       date          date1         flag
 1       2010-05-08     2010-05-09    1 
 2       2012-08-08     2012-08-08    0 
 3       2013-06-23     <NA>          NA

DATA.TABLE

library(data.table)

data_2 <- merge.data.table(A, B, by = "customer", all.x=TRUE)
data_2[, logic:= (ifelse(as.Date(data$application_date.x) + 1 == as.Date(data$application_date.y), 1, 0))]
setnames(data_2, old = c("customer", "application_date.x", "application_date.y", "logic"), 
                 new = c("id", "date", "date1", "flag"))

输出

 id       date          date1         flag
 1       2010-05-08     2010-05-09    1 
 2       2012-08-08     2012-08-08    0 
 3       2013-06-23     <NA>          NA

答案 1 :(得分:1)

如果您不介意直接通过引用更新A,则可以使用以下更新联接选项:

library(data.table)
setDT(A)[, d := dateA + 1L]
setDT(B)
A[B, on=.(id, d=dateB), c("dateB", "flag") := .(dateB, 1L)]

数据:

A <- data.frame(id = c(1,2,3),
    dateA = as.Date(c("2010-05-08", "2012-08-08", "2013-06-23")))
B <- data.frame(id = c(1,2),
    dateB = as.Date(c("2010-05-09", "2012-08-08")))

答案 2 :(得分:0)

使用data.table:

data <- merge(A, B, by = "customer",all.x = TRUE)
library(data.table)

setDT(data)
data[as.Date(application_date.x)+1==application_date.y,flag:=1]
data[as.Date(application_date.x)+1!=application_date.y,flag:=0]
data <- data[,.(id=customer, date=application_date.x,date1=application_date.y,flag)]
data