使用数据中的行操作创建新列。表R

Question

我在R中有两个数据表，如下:

DT_A具有列sid、date、value e1、value e2，其中sid是主键，date是次要键。

DT_B有列sid、date1、date2、date3、......、date12。这里sid是主键，因此每一行对应于唯一的sid，其他列名对应于DT_A中的辅助键。

我想在< code>DT_A中添加另一列，其中第I行包含表DT_B[sid_i，date_i]中的相应元素

下面是示例代码和所需输出：

require(data.table)
DT_A <- data.table(sid = c(1,2,3,4,5,1,3), 
                   date = c("Jan 2012", "Feb 2012", "April 2012", "May 2012", 
                            "Dec 2012", "Feb   2012", "Oct 2012"),
                   value1 = rep("1", 7), 
                   value2 = rep("1", 7))

DT_B <- data.table(sid = as.character(c(1,2,3,4,5)), 
                   "Jan 2012" = rep("1", 5),
                   "Feb 2012" = rep("2", 5),
                   "March 2012" = rep("3", 5),
                   "April 2012" = rep("4", 5),
                   "May 2012" = rep("5", 5),
                   "June 2012" = rep("6", 5),
                   "July 2012" = rep("7", 5),
                   "Aug          2012" = rep("8", 5),
                   "Sept 2012" = rep("9", 5),
                   "Oct 2012" = rep("10", 5),
                   "Nov 2012" = rep("11", 5),
                   "Dec 2012" = rep("12", 5))

#Set Keys
setkey(DT_A, sid, date)
setkey(DT_B, sid)

#Define Function fun1
fun1 <- function(x){ 
    tryCatch(DT_B[x[1], x[2], with=FALSE], error = function(e) NULL)
}

#Desired Output
DT_A$newCol <- sapply(apply(DT_A, 1, fun1),"[[",1)
DT_A

虽然我当前的方法在这个小例子上有效，但我的实际DT_A有2000万行，这个方法完全挂在那里，我们可以使用data.table或任何其他方法以更有效的方式进行此操作吗？

Answer 1

熔化第二个＜code＞数据。表：

library(reshape2)

DT_B.melted = melt(DT_B, id.vars = 'sid', variable.name = 'date')

一旦它融化了，你就可以把两者设置成同一个键，然后做你的join/assign/whatnot:

setkey(DT_B.melted, sid, date)
setkey(DT_A, sid, date)

DT_A[DT_B.melted, newCol := value]