celestialgod

spread sort column names

Aug 7th, 2015
384
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
R 2.57 KB | None | 0 0
  1. library(data.table)
  2. library(dplyr)
  3. library(tidyr)
  4. library(magrittr)
  5.  
  6. DT = data.table(stu = paste0("stu", 1:20),
  7.   X = rnorm(20), Y = rnorm(20)) %>%
  8.   gather(cate, values, -stu)
  9. DT %>% spread(stu, values) %>% tbl_dt(FALSE)
  10. #   cate        stu1      stu10     stu11      stu12       stu13     stu14
  11. # 1    X -0.08476976  0.5428922 1.9929332 -0.6145632 -0.06098296 -1.066283
  12. # 2    Y  0.59710869 -1.0037766 0.3508158  0.4587201 -0.13639207  1.385517
  13. # Variables not shown: stu15 (dbl), stu16 (dbl), stu17 (dbl), stu18 (dbl),
  14. #   stu19 (dbl), stu2 (dbl), stu20 (dbl), stu3 (dbl), stu4 (dbl), stu5 (dbl),
  15. #   stu6 (dbl), stu7 (dbl), stu8 (dbl), stu9 (dbl)
  16.  
  17. ## factorize
  18. DT = data.table(stu = paste0("stu", 1:20),
  19.   X = rnorm(20), Y = rnorm(20)) %>%
  20.   gather(cate, values, -stu)
  21. DT %>% mutate(stu = factor(stu, levels = paste0("stu", 1:20))) %>%
  22.   spread(stu, values) %>% tbl_dt(FALSE)
  23. #   cate      stu1      stu2      stu3      stu4       stu5       stu6
  24. # 1    X 1.6890231 -1.300332 -1.378376 -1.874321 0.54141060 -1.2848391
  25. # 2    Y 0.2796895  1.635385  1.048334  0.424909 0.09111916 -0.4147811
  26. # Variables not shown: stu7 (dbl), stu8 (dbl), stu9 (dbl), stu10 (dbl), stu11
  27. #   (dbl), stu12 (dbl), stu13 (dbl), stu14 (dbl), stu15 (dbl), stu16 (dbl),
  28. #   stu17 (dbl), stu18 (dbl), stu19 (dbl), stu20 (dbl)  
  29.  
  30. ## method 1 with select
  31. DT_spread = DT %>% spread(stu, values)
  32. sele_names = setdiff(names(DT_spread), paste0("stu", 1:20))
  33. col_num = match(c(sele_names, paste0("stu", 1:20)), names(DT_spread))
  34. DT_spread %>% select(col_num) %>% tbl_dt(FALSE)
  35. #   cate      stu1      stu2      stu3      stu4       stu5       stu6
  36. # 1    X 1.6890231 -1.300332 -1.378376 -1.874321 0.54141060 -1.2848391
  37. # 2    Y 0.2796895  1.635385  1.048334  0.424909 0.09111916 -0.4147811
  38. # Variables not shown: stu7 (dbl), stu8 (dbl), stu9 (dbl), stu10 (dbl), stu11
  39. #   (dbl), stu12 (dbl), stu13 (dbl), stu14 (dbl), stu15 (dbl), stu16 (dbl),
  40. #   stu17 (dbl), stu18 (dbl), stu19 (dbl), stu20 (dbl)
  41.  
  42.  
  43. ## method 2
  44. DT %<>% mutate(stu_num = as.integer(gsub("stu(\\d*)", "\\1", stu)),
  45.   stu = sprintf("stu%02i", stu_num)) %>% select(-stu_num)
  46. DT %>% spread(stu, values) %>% tbl_dt(FALSE)
  47. #   cate     stu01     stu02     stu03     stu04      stu05      stu06
  48. # 1    X 1.6890231 -1.300332 -1.378376 -1.874321 0.54141060 -1.2848391
  49. # 2    Y 0.2796895  1.635385  1.048334  0.424909 0.09111916 -0.4147811
  50. # Variables not shown: stu07 (dbl), stu08 (dbl), stu09 (dbl), stu10 (dbl),
  51. #   stu11 (dbl), stu12 (dbl), stu13 (dbl), stu14 (dbl), stu15 (dbl), stu16
  52. #   (dbl), stu17 (dbl), stu18 (dbl), stu19 (dbl), stu20 (dbl)
Advertisement
Add Comment
Please, Sign In to add comment