2016-05-05 6 views
0

ich einen Datenrahmen haben („df“) von zwei Variablen x und y:dcast dann transformierten Variablen concatenate

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 

Was ich ist es, die Datenrahmen reduzieren jetzt tun möchte nur verschiedene x-Variablen enthalten (zB eine "1", eine "2", eine "3" usw.) und haben ein neues Feld, das die einzelnen x-Werte miteinander verknüpft, die durch Kommata getrennt sind. Das Endergebnis würde wie folgt aussehen:

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 

ich eine Lösung habe jetzt das funktioniert, aber ich denke, es ist umständlicher als nötig. Meine aktuelle Lösung ist unten und für einen "df" -Datenrahmen von 5.000 Zeilen dauert es etwa 12 Sekunden, um "df3" zu erstellen.

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)))) 

Mein eigentlicher Datenrahmen ist mehr als 80.000 Zeilen lang, und ich habe diese Art von Funktion 4 verschiedener Zeiten in meinem Programm auszuführen.

Ich freue mich über jeden Rat, den Sie mir geben können, um diesen Prozess zu beschleunigen.

+1

Sie können versuchen, 'Bibliothek (dplyr); df%>% group_by (x)%>% zusammenfassen (y = toString (y)) ' –

Antwort

2

Wie wäre es,

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