创建含table()函数的R自定义函数时报参数长度一致错误如何解决

错误原因 #

报错核心是$运算符的解析规则导致的：在自定义函数中使用df$var时，R会直接在数据框中查找字面名为var的列，而非解析你传入函数的var参数对应的实际列名。你的示例数据中没有叫var的列，所以df$var返回NULL，和长度为4的df$arm长度不一致，因此触发报错。

修复方案 #

方法1：用双括号[[解析变量名（通用基础方案） #

如果要传入未加引号的列名，先把参数转换为字符串再取列即可：

f.table <- function(df, var) {
  # 将传入的未加引号列名转为字符串
  var_name <- deparse(substitute(var))
  p <- table(df[[var_name]], df$arm) %>%
    fisher.test(alternative = "two.sided") %>%
    .$p.value
  return(p)
}

# 调用后可正常返回计算结果
f.table(study.df, x)

方法2：传入引号包裹的列名 #

如果可以接受传字符串形式的列名，代码可以更简化：

f.table <- function(df, var) {
  p <- table(df[[var]], df$arm) %>%
    fisher.test(alternative = "two.sided") %>%
    .$p.value
  return(p)
}

# 调用时传入列名字符串即可
f.table(study.df, "x")

方法3：tidyverse风格适配方案 #

如果习惯使用tidyverse语法，用rlang的卷曲运算符{{}}解析参数即可：

library(dplyr)

f.table <- function(df, var) {
  p <- table(pull(df, {{var}}), df$arm) %>%
    fisher.test(alternative = "two.sided") %>%
    .$p.value
  return(p)
}

f.table(study.df, x)

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