使用R函数中的paste()在as.formula()中的table()

时间:2017-11-24 10:46:00

标签: r function chi-squared

我正在使用以下数据集

hsb2 <- within(read.csv("https://stats.idre.ucla.edu/stat/data/hsb2.csv"), {
  race <- as.factor(race)
  schtyp <- as.factor(schtyp)
  prog <- as.factor(prog)
})

我希望编写一个高阶函数来对分类依赖变量(如卡方或Fisher)进行统计检验。

作为一个例子,我正在运行这两个测试:

chisq.test(table(hsb2$female, hsb2$schtyp))
chisq.test(table(hsb2$race, hsb2$schtyp))

我将高阶函数定义为:

multi.tests.categorical <- function(fun = chisq.test, df, vars, group.var, ...) {
      sapply(simplify = FALSE,                                    
             vars,                                                
             function(var) {
               formula <- as.formula(paste("table(",var, ",", group.var,")")) # create a formula with outcome and grouping var.
               fun(data = df, formula, ...)                     # perform test with a given fun, default t.test
             }
      )
    }

然后我在一个代码块中应用我的高阶函数:

res.multi.chisq.tests <-
  multi.tests.categorical(fun = chisq.test,
              df = hsb2,
              vars = "schtyp",
              group.var = c("female","race"))
res.multi.chisq.tests

但是,我收到以下错误消息:

  

表中的错误(schtyp,female):object&#39; schtyp&#39;找不到

我怀疑我在as.formula中使用table()可能是原因造成的?任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:1)

Vectorize解决方案可能是

#sample data
hsb2 <- within(read.csv("https://stats.idre.ucla.edu/stat/data/hsb2.csv"), {
  race <- as.factor(race)
  schtyp <- as.factor(schtyp)
  prog <- as.factor(prog)
})

schtyp_idx <- match("schtyp", colnames(hsb2))
col_idx <- match(c("female", "race"), colnames(hsb2))
chisq.test_resultList = mapply(function(x,y){chisq.test(table(hsb2[,x],hsb2[,y]))}, schtyp_idx, col_idx)
chisq.test_resultList