使用plyr(或其他)使用来自2个其他变量的seq创建变量的更好方法

时间:2013-04-19 07:21:59

标签: r plyr

我有一些包含入场和出院日期的数据,我需要(对于每个人并且承认ID#),创建一个包含人员和承认ID的宽到长的数据框,其中包含承认和解除,表明其间的每一天的价值,包括承认和解除。

我偶然发现了一种非常黑客的方法,它利用dlply如何命名它创建的列表中的项目;但是,我担心可能容易出错。无论哪种方式,它都感觉很笨拙,我想知道是否有更简洁的方法来使用更少的代码。

person <- c(1, 2, 3, 3)
admit <- c(1, 1, 1, 2)
admit.date <- as.Date(c("1/1/2010", "1/1/2010", "1/1/2010", "2/1/2010"), "%m/%d/%Y")
discharge.date <- as.Date(c("1/1/2010",  "1/1/2010", "1/1/2010", "2/1/2010"), "%m/%d/%Y") + c(1,2,2,2)
df1 <- data.frame(person, admit, admit.date, discharge.date)
df1 ## where I start

library(plyr)
los_seq <-  function(df) { seq(df$admit.date, df$discharge.date, 1)} 
lst1 <- dlply(df1, .(person, admit), los_seq)
vec1 <- unlist(lst1)
## now it gets really hackish
df2 <- data.frame(v1 = paste(names(vec1), vec1, sep="__"))
df2$person <- substr(df2$v1, 1, regexpr("\\.", df2$v1)-1)
df2$admit <- substr(df2$v1, regexpr("\\.", df2$v1)+1,  regexpr("\\.", df2$v1)+1)
df2$date <- as.Date(as.numeric(substr(df2$v1, regexpr("__", df2$v1)+2,  nchar(df2$v1))), origin="1970-01-01")
df2[,-1]  ##  this is how I need the result to look

1 个答案:

答案 0 :(得分:2)

如何使用以下内容(因为您使用的是实际日期):

SEQ <- df1$discharge.date - df1$admit.date + 1
df1[rep(row.names(df1), SEQ), 1:2]
#     person admit
# 1        1     1
# 1.1      1     1
# 2        2     1
# 2.1      2     1
# 2.2      2     1
# 3        3     1
# 3.1      3     1
# 3.2      3     1
# 4        3     2
# 4.1      3     2
# 4.2      3     2

现在,想一想你在评论中可能提到的内容,或许你也在寻找日期序列。

SEQ <- df1$discharge.date - df1$admit.date + 1
df2 <- df1[rep(row.names(df1), SEQ), 1:3]
df2$date <- df2$admit.date + sequence(SEQ)-1
df2[-3]
#     person admit       date
# 1        1     1 2010-01-01
# 1.1      1     1 2010-01-02
# 2        2     1 2010-01-01
# 2.1      2     1 2010-01-02
# 2.2      2     1 2010-01-03
# 3        3     1 2010-01-01
# 3.1      3     1 2010-01-02
# 3.2      3     1 2010-01-03
# 4        3     2 2010-02-01
# 4.1      3     2 2010-02-02
# 4.2      3     2 2010-02-03