Цитируя ад с формулами

Question

У меня есть задача, которую мне нужно выполнять несколько раз, поэтому я пытаюсь написать для нее функцию. Моя функция почти завершена, за исключением одного дефекта: я не могу передать формулу в качестве аргумента, если она не в форме строки.

library(lme4)
library(lazyeval)

get_pvals_qrhs <- function(df, cols, qrhs) {
    cols <- substitute(cols)
    col_pos <- setNames(as.list(seq_along(df)), names(df))
    pos <- eval(cols, col_pos)
    formulas <- lapply(pos, function(x) formula(paste(colnames(df[x]), "~", qrhs)))
    models <- lapply(formulas, lmer, data=df, REML=FALSE)
    tvals <- lapply(models, function(x) data.frame(coef(summary(x)))[c(2), ]$t.value)
    pvals <- lapply(tvals, function(x) { 2 * (1 - pnorm(abs(x))) })
    return(unlist(pvals))
}

works <- get_pvals_qrhs(iris, Sepal.Length:Sepal.Width, "Species + (1 + Petal.Length | Petal.Width)")
works

get_pvals_rhs <- function(df, cols, rhs) {
    cols <- substitute(cols)
    col_pos <- setNames(as.list(seq_along(df)), names(df))
    pos <- eval(cols, col_pos)
    formulas <- lapply(pos, function(x) formula(paste(colnames(df[x]), "~", quote(rhs))))
    models <- lapply(formulas, function(x) lmer, data=df, REML=FALSE)
    tvals <- lapply(models, function(x) data.frame(coef(summary(x)))[c(2), ]$t.value)
    pvals <- lapply(tvals, function(x) { 2 * (1 - pnorm(abs(x))) })
    return(unlist(pvals))
}

fails <- get_pvals_rhs(iris, Sepal.Length:Sepal.Width, Species + (1 + Petal.Length | Petal.Width))
fails

Независимо от того, что я делаю с термином rhs в строке formulas <- ... во второй функции, я не могу получить результаты, полученные от первой функции. Что я делаю не так?

Answer 1

Вам нужно заменить, чтобы получить выражение, и затем вы можете отменить его, чтобы превратить это в строкус частями формул обычно лучше всегда ставить перед ними префикс ~, поэтому вам не нужно вводить substitute(), и вы можете использовать такие функции, как update(), чтобы упростить изменение формул.Нечто подобное

get_pvals_rhs <- function(df, cols, rhs) {
  cols <- substitute(cols)
  col_pos <- setNames(as.list(seq_along(df)), names(df))
  pos <- eval(cols, col_pos)
  formulas <- lapply(colnames(df[pos]), function(x) update(rhs, reformulate(".", x)))
  models <- lapply(formulas, lmer, data=df, REML=FALSE)
  tvals <- lapply(models, function(x) data.frame(coef(summary(x)))[c(2), ]$t.value)
  pvals <- lapply(tvals, function(x) { 2 * (1 - pnorm(abs(x))) })
  return(unlist(pvals))
}
get_pvals_rhs(iris, Sepal.Length:Sepal.Width, ~ Species + (1 + Petal.Length | Petal.Width))