我正在处理与下面类似的汇总数据集,我需要对其进行扩展,使其看起来像第二个数据集。
df <- data.frame(CustName = letters[1:3],
Years = c(4,2,1),
MinYear = c(1995,1992,1998),
stringsAsFactors = F)
df
我试过使用循环,但我没有成功
所需的输出是这样的
dfResult <- data.frame(CustName = rep(letters[1:3], c(4,2,1)),
Years = c(1995:1998, 1992:1993, 1998), stringsAsFactors = F)
dfResult
答案 0 :(得分:5)
您基本上需要按客户名称拆分数据集,然后根据每个客户的数据创建新的数据框。我们通过在开始年份添加0:(Years-1)来完成此操作。 -1表示起始年份。最后,我们将这些绑定在一起。我们可以在base-R中执行此操作:
res <- do.call(rbind,lapply(split(df,df$CustName),function(x){
res <- data.frame(custName=x$CustName,
Year=x$MinYear+0:(x$Years-1))
res
}))
# > res
# custName Year
# a.1 a 1995
# a.2 a 1996
# a.3 a 1997
# a.4 a 1998
# b.1 b 1992
# b.2 b 1993
# c c 1998
我们可以对data.table做同样的事情,使代码更具可读性:
library(data.table)
DT <- as.data.table(df)
res <- DT[,.(Year=MinYear+0:(Years-1)),CustName]