基于R中的条件操纵向量

时间:2018-01-04 06:46:31

标签: r bigdata q sapply

我正在尝试找到一种有效的方法来执行以下代码片段,但是在相当大的数据帧(2M +记录)上。我已尝试使用sapplyfor loop,但两者都不是很好。

' ACCOUNT_CLOSE_DATE'恰好是空白但它是最后一个' AS_OF_DATE'对于给定的' U_ID'我想将此特定值插入' ACCOUNT_CLOSE_DATE' ' U_ID'的列。任何帮助将不胜感激!

 library(reshape2)
 cnames = c("Date", "999:999", "888:888", "777:777")
 dates = c("31JAN2005", "28FEB2005", "31MAR2005")
 val = ""
 mydf = data.frame(cbind(dates, matrix(val, nrow = 3, ncol = 3)))
 colnames(mydf) = cnames
 nmydf = melt(mydf, id.vars = "Date")
 nmydf$Date = as.character(nmydf$Date)
 colnames(nmydf) = c("AS_OF_DATE", "U_ID", "ACCOUNT_CLOSE_DATE")
 uvals = c("999:999", "888:888", "777:777")
 nmydf[nmydf$U_ID == uvals[1],"ACCOUNT_CLOSE_DATE"] = 
   nmydf[nmydf$U_ID == uvals[1], "AS_OF_DATE"][length(nmydf[nmydf$U_ID == uvals[1], "AS_OF_DATE"])]

1 个答案:

答案 0 :(得分:0)

您可以执行data.table联接,以便将每组“U_ID”的结束日期输入到表中,如下所示:

library(data.table)
cnames <-  c("Date", "999:999", "888:888", "777:777")
dates <-  c("31JAN2005", "28FEB2005", "31MAR2005")
val <-  ""
mydt <-  data.table(cbind(dates, matrix(val, nrow = 3, ncol = 3)))
setnames(mydt, cnames)
nmydt <-  melt(mydt, id.vars = "Date")[,Date:variable]
setnames(nmydt, c("AS_OF_DATE", "U_ID"))
closing_dates <- setnames(nmydt[, .SD[.N], by=U_ID], c("U_ID", "ACCOUNT_CLOSE_DATE"))
merge(nmydt, closing_dates, by="U_ID", all.x=TRUE)

#       U_ID AS_OF_DATE ACCOUNT_CLOSE_DATE
# 1: 999:999  31JAN2005          31MAR2005
# 2: 999:999  28FEB2005          31MAR2005
# 3: 999:999  31MAR2005          31MAR2005
# 4: 888:888  31JAN2005          31MAR2005
# 5: 888:888  28FEB2005          31MAR2005
# 6: 888:888  31MAR2005          31MAR2005
# 7: 777:777  31JAN2005          31MAR2005
# 8: 777:777  28FEB2005          31MAR2005
# 9: 777:777  31MAR2005          31MAR2005

此解决方案还应提高效率。