如何在lme4中展开单竖线随机效应公式？

一、通用交互项展开函数 #

要把语言对象如x * w展开为x + w + x:w，可以利用R公式系统的terms()函数自动解析交互项的特性，实现一个通用函数：

expand_interaction <- function(expr) {
  # 把表达式转为公式形式
  f <- formula(paste0("~", deparse(expr)))
  # 提取所有解析后的项（去掉截距标记）
  term_labels <- attr(terms(f), "term.labels")
  # 重新拼接为表达式返回
  parse(text = paste(term_labels, collapse = " + "))[[1]]
}

测试示例：

expand_interaction(quote(x * w))
# 返回：x + w + x:w

二、处理单竖线随机效应公式 #

针对(1 + x * w | z)这类单竖线随机效应公式，我们可以拆分公式结构，展开左侧的交互项后重新组合：

expand_single_vert <- function(formula) {
  # 定位随机效应部分
  rand_component <- formula[[length(formula)]]
  # 拆分竖线左右两侧的内容
  left_side <- rand_component[[2]]
  right_side <- rand_component[[3]]
  
  # 展开左侧的交互项
  expanded_left <- if (is.call(left_side)) {
    if (left_side[[1]] == quote(`+`)) {
      # 遍历每个项，展开其中的交互项
      expanded_terms <- lapply(left_side[-1], function(term) {
        if (is.call(term) && term[[1]] == quote(`*`)) {
          expand_interaction(term)
        } else {
          term
        }
      })
      as.call(c(quote(`+`), expanded_terms))
    } else if (left_side[[1]] == quote(`*`)) {
      expand_interaction(left_side)
    } else {
      left_side
    }
  } else {
    left_side
  }
  
  # 重新构建随机效应部分并组合成完整公式
  new_rand <- as.call(list(quote(`|`), expanded_left, right_side))
  as.call(list(formula[[1]], formula[[2]], new_rand))
}

测试示例：

# 原始公式
orig_form <- formula(y ~ (1 + x * w | z))
# 展开后的公式
expand_single_vert(orig_form)
# 返回：y ~ (1 + x + w + x:w | z)

原理说明 #

R的terms()函数会按照公式语法规则自动解析*运算符，将其展开为所有主效应与交互效应的组合，这是R公式系统的原生能力，我们只需利用这个特性提取解析后的项，再重新构建所需的表达式或公式即可。

内容的提问来源于stack exchange，提问作者Jigsaw