然后dcast连接变换后的变量

时间:2016-05-05 12:53:52

标签: r dataframe reshape2

我有两个变量x和y的数据帧(" df"):

a <- c(1:2000)
b <- c(1:4000)
c <- 5000

df <- data.frame(x = sample(a, c, replace = TRUE),
                 y = as.character(sample(b, c, replace = TRUE)))
df <- df[order(df$x), ]

head(df, 10)
     x    y
881  1 2919
4425 1 2000
2478 2 3375
4808 2 3928
4871 2 3351
4889 2 1634
1242 3 3957
1378 3 3356
3029 3 2625
3657 4  646

我现在要做的是将数据帧减少为仅包含不同的x变量(例如,一个&#34; 1&#34;,一个&#34; 2&#34;,一个&#34; ; 3&#34;等)并且有一个新字段,用于连接每个不同的x&sy; sy值,用逗号分隔。最终结果如下:

head(df3)
  x               multi_ys
1 1             2000, 2919
2 2 1634, 3351, 3375, 3928
3 3       2625, 3356, 3957
4 4              1092, 646
5 5   113, 2430, 3187, 932
6 7                   2349

我现在有一个有效的解决方案,但我认为它比必要的更麻烦。我的当前状态解决方案如下所示,对于&#34; df&#34;数据帧为5,000行,创建&#34; df3&#34;需要大约12秒。

library(reshape2)

#creates a duplicate field of y, to dcast in the 'multi_y' function below
df$y2 <- df$y

#creates a new dataframe with unique x values
df2 <- df[which(!duplicated(df$x)), ]


multi_y <- function(x) {
  c3.i <- df2[x, 1]
  c3.j <- df[df$x == c3.i, ]
  c3.k <- dcast(c3.j, x ~ y, value.var = "y2")
  cols <- colnames(c3.k)

  #if there are more than two columns in this loop's data frame, then concatenate all columns except the first
  if(ncol(c3.k) > 2) {
    c3.k$cycles <- apply( c3.k[ , cols[-1]] , 1 , paste , collapse = ", " )
  } else {
    c3.k$cycles <- c3.k[, 2]
  }

  c3.l <- cbind(data.frame(c3.k[, 1]), data.frame(c3.k[, ncol(c3.k)]))
  colnames(c3.l) <- c("x", "multi_ys")
  print(c3.l)
}

t <- (1:nrow(df2))
system.time(df3 <- do.call("rbind", lapply(t, function(x) multi_y(x))))

我的实际数据帧超过80,000行,我必须在程序中运行此类函数4次。

我感谢您提供的任何建议,以帮助我加快这一过程。

1 个答案:

答案 0 :(得分:2)

怎么样,

df1 <- aggregate(y~x, df, paste, collapse = ',')
head(df1)
#  x                                   y
#1 1                                 542
#2 2                      3813,1220,1666
#3 3 1713,35,643,3957,872,2235,3015,3051
#4 4                      2037,1371,1180
#5 5                            2724,905
#6 6                            293,3248