Reshape wide format, to multi-column long format

前端 未结 4 1635
一生所求
一生所求 2020-12-14 10:01

I want to reshape a wide format dataset that has multiple tests which are measured at 3 time points:

   ID   Test Year   Fall Spring Winter
    1   1   2008          


        
4条回答
  •  温柔的废话
    2020-12-14 10:24

    Using reshape2:

    # Thanks to Ista for helping with direct naming using "variable.name"
    df.m <- melt(df, id.var = c("ID", "Test", "Year"), variable.name = "Time")
    df.m <- transform(df.m, Test = paste0("Test", Test))
    dcast(df.m, ID + Year + Time ~ Test, value.var = "value")
    

    Update: Using data.table melt/cast from versions >= 1.9.0:

    data.table from versions 1.9.0 imports reshape2 package and implements fast melt and dcast methods in C for data.tables. A comparison of speed on bigger data is shown below.

    For more info regarding NEWS, go here.

    require(data.table) ## ver. >=1.9.0
    require(reshape2)
    
    dt <- as.data.table(df, key=c("ID", "Test", "Year"))
    dt.m <- melt(dt, id.var = c("ID", "Test", "Year"), variable.name = "Time")
    dt.m[, Test := paste0("Test", Test)]
    dcast.data.table(dt.m, ID + Year + Time ~ Test, value.var = "value")
    

    At the moment, you'll have to write dcast.data.table explicitly as it's not a S3 generic in reshape2 yet.


    Benchmarking on bigger data:

    # generate data:
    set.seed(45L)
    DT <- data.table(ID = sample(1e2, 1e7, TRUE), 
            Test = sample(1e3, 1e7, TRUE), 
            Year = sample(2008:2014, 1e7,TRUE), 
            Fall = sample(50, 1e7, TRUE), 
            Spring = sample(50, 1e7,TRUE), 
            Winter = sample(50, 1e7, TRUE))
    DF <- as.data.frame(DT)
    

    reshape2 timings:

    reshape2_melt <- function(df) {
        df.m <- melt(df, id.var = c("ID", "Test", "Year"), variable.name = "Time")
    }
    # min. of three consecutive runs
    system.time(df.m <- reshape2_melt(DF))
    #   user  system elapsed 
    # 43.319   4.909  48.932 
    
    df.m <- transform(df.m, Test = paste0("Test", Test))
    
    reshape2_cast <- function(df) {
        dcast(df.m, ID + Year + Time ~ Test, value.var = "value")
    }
    # min. of three consecutive runs
    system.time(reshape2_cast(df.m))
    #   user  system elapsed 
    # 57.728   9.712  69.573 
    

    data.table timings:

    DT_melt <- function(dt) {
        dt.m <- melt(dt, id.var = c("ID", "Test", "Year"), variable.name = "Time")
    }
    # min. of three consecutive runs
    system.time(dt.m <- reshape2_melt(DT))
    #   user  system elapsed 
    #  0.276   0.001   0.279 
    
    dt.m[, Test := paste0("Test", Test)]
    
    DT_cast <- function(dt) {
        dcast.data.table(dt.m, ID + Year + Time ~ Test, value.var = "value")
    }
    # min. of three consecutive runs
    system.time(DT_cast(dt.m))
    #   user  system elapsed 
    # 12.732   0.825  14.006 
    

    melt.data.table is ~175x faster than reshape2:::melt and dcast.data.table is ~5x than reshape2:::dcast.

提交回复
热议问题