2015-06-30 3 views
5

Desidero correggere una tabella in base alle sostituzioni in un'altra tabella. Voglio cambiare il valore in dt_current quando dt_override ha quell'unità e gli intervalli di date si sovrappongono a dt_current.Unire due tabelle dati per sovrascrivere i valori per intervallo di date

dt_current <- data.table(unit = c(rep("a",10), rep("b", 10)), 
    date = seq(as.Date("2015-1-1"), by = "day", length.out = 10), 
    num = 1:10, key = "unit") 
dt_override <- data.table(unit = c("a", "a", "b", "zed"), start_date = as.Date(c("2015-01-03", "1492-12-25", "2015-01-02", "2015-01-11")), 
    end_date = as.Date(c("2015-01-05", "1492-12-26", "2015-01-04", "2015-01-14")), 
    value = NA, key = "unit") 

sembra che dovrei usare una qualche forma di .EACHI quando si uniscono le due tabelle di dati, codificati qualcosa di simile a quanto segue, ho pensato che non funziona o corso.

dt_current[dt_override, 
    num := if(i.start_date <= date & i.end_date >= date) i.value, 
    by = .EACHI] 

risposta

6

Utilizzando foverlaps si potrebbe fare

dt_current[, date2 := date] # define end date 
setkey(dt_current, unit, date, date2) # key by unit, start and end dates 
setkey(dt_override, unit, start_date, end_date) # same 

Prima opzione, creare e indice e aggiornamento con riferimento

indx <- foverlaps(dt_override, dt_current, which = TRUE) # run foverlaps and get indices 
dt_current[indx$yid, num := dt_override[indx$xid, value]] # adjust by reference 

Alernatively, è possibile eseguire foverlaps il contrario e evitare di creare indx ma durante la creazione di un intero nuovo set di dati

foverlaps(dt_current, dt_override)[!is.na(start_date), num := value 
            ][, names(dt_current), with = FALSE] 
2

Ecco un modo, enumerando le sequenze di data:

dt_override[,value:=as.integer(value)] 
# It's necessary to convert to integer because `NA` is logical unless otherwise specified. 

dto = dt_override[,.(
    unit, 
    date = seq.Date(start_date,end_date,by="day"), 
    value 
),by=seq_along(dt_override)][,seq_along:=NULL] 

setkey(dt_current,unit,date) 
dt_current[dto,num:=i.value] 

Ora che foverlaps è in, probabilmente c'è un modo molto migliore.

4

Un'altra alternativa, utilizzando laminazione unisce:

setkey(dt_current, unit, date) 
setkey(dt_override, unit, start_date) 

dt_current[, num := dt_override[dt_current, roll = T][end_date >= start_date, 
                 num := value]$num] 

# another version of the above, but using ifelse (unclear to me which one is faster) 
dt_current[, num := dt_override[dt_current, 
           ifelse(end_date >= start_date, value, num), roll = T]] 
+0

che soffia un pò la mia mente –