如何在不同分类变量的水平上应用Chisq.test?

时间:2015-06-12 07:02:42

标签: r statistics sapply chi-squared

我想在分类变量的每个级别上执行chisq.test()

目前,我已设法使用以下代码对每个分类变量执行此操作。

# Random generation of values for categorical data
set.seed(12)
x <- data.frame(col1 = sample( LETTERS[1:4], 100, replace=TRUE ), 
                col2 = sample( LETTERS[3:6], 100, replace=TRUE ),
                col3 = sample( LETTERS[2:5], 100, replace=TRUE ),
                out = sample(c(1,2),100, replace=TRUE))

# performing chisq.test
pval <- as.data.frame(sapply(c(1:3),function(i)chisq.test(x[,i],x[,'out'])$p.value ))

#output
    p.value
1 0.33019256
2 0.08523487
3 0.79403367

我有兴趣比较不同结果的水平。

# for col1 levels different outcomes
table(x$col1,x$out)

#output
     1  2
  A  8 12
  B 18 10
  C 12 11
  D 18 11

例如,要将col1中的B级与out中的不同结果1,2进行比较。

我想知道如何将这个(或以另一种智能方式)扩展到分类变量的每个级别?

# Expected output
       p.value

col1.A  *****
col1.B  *****
col1.C  *****
.
.
.
col3.E  *****

感谢您的关注。

1 个答案:

答案 0 :(得分:0)

如果你想对给定的概率进行卡方检验(p = rep(0.5, 2)),你可以这样做。

我已将其分解为易于理解:

getP <- function(lev, x, i) {
  tab <- table(x$out[x[, i] == lev])
  chisq.test(tab)$p.value
}
pvalList <- lapply(1:3, function(i) {
  df <- data.frame(Column = i, Category = levels(x[, i]))
  df$p.value <- sapply(df$Category, getP, x, i)
  df
})
pval <- do.call("rbind", pvalList) # Convert to single data frame

或者,如果您想要的实际上是A而不是A,B vs而不是B等,您可以用getP的定义替换:

getP <- function(lev, x, i) {
  tab <- table(x$out, x[, i] == lev)
  chisq.test(tab)$p.value
}