使用R中的循环重复值

时间:2016-01-05 16:56:53

标签: r loops repeat

我正在处理与下面类似的汇总数据集,我需要对其进行扩展,使其看起来像第二个数据集。

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

1 个答案:

答案 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]