В функции R: dcast передайте имена столбцов (снова!)

Дано df в полудлинном формате с переменными id a а также b и измеренные данные в столбцах m1а также m2, Тип данных определяется переменной v (значения var1 и var2).

set.seed(8)

df_l <- 
  data.frame(
    a = rep(sample(LETTERS,5),2),
    b = rep(sample(letters,5),2),
    v = c(rep("var1",5),rep("var2",5)),
    m1 = sample(1:10,10,F),
    m2 = sample(20:40,10,F)) 

Выглядит как:

   a b    v m1 m2
1  W r var1  3 40
2  N l var1  6 32
3  R a var1  9 28
4  F g var1  5 21
5  E u var1  4 38
6  W r var2  1 35
7  N l var2  8 33
8  R a var2 10 29
9  F g var2  7 30
10 E u var2  2 23

Если я хочу сделать широкий формат значений в m1 используя идентификатор a в виде строк и значений в v1в качестве столбцов я делаю:

> reshape2::dcast(df_l, a~v, value.var="m1")
  a var1 var2
1 E    4    2
2 F    5    7
3 N    6    8
4 R    9   10
5 W    3    1

Как мне написать функцию, которая делает это аргументами dcast (row, column и value.var) предоставляются в качестве аргументов, что-то вроде:

fun <- function(df,row,col,val){
  require(reshape2)
  res <-
    dcast(df, row~col, value.var=val)
  return(res)
}

Я проверил ТАК здесь и здесь, чтобы попробовать варианты match.call а также eval(substitute()) чтобы "получить" аргументы внутри функции, а также попытался с помощью пакета lazyeval. Нет успеха.

Что я здесь не так делаю? Как заставить dcast распознавать имена переменных?

1 ответ

Решение

Аргумент формулы также принимает ввод символов.

foo <- function(df, id, measure, val) {
    dcast(df, paste(paste(id, collapse = " + "), "~", 
                    paste(measure, collapse = " + ")), 
          value.var = val)
}

require(reshape2)
foo(df_l, "a", "v", "m1")

Обратите внимание, что data.table"s dcast ( текущая разработка) также может привести несколько value.var колонки напрямую. Итак, вы также можете сделать:

require(data.table) # v1.9.5
foo(setDT(df_l), "a", "v", c("m1", "m2"))
#    a m1_var1 m1_var2 m2_var1 m2_var2
# 1: F       1       6      28      21
# 2: H       9       2      38      29
# 3: M       5      10      24      35
# 4: O       8       3      23      26
# 5: T       4       7      31      39
Другие вопросы по тегам