`все наблюдения находятся в одной группе`, когда` bartlett.test`

Question

variance_homo<-function(df,col1,col2){
    if (is.numeric(df[,col2]) & nlevels(df[,col1])>2){
    bartlett_res<-bartlett.test(col2~col1,df,na.action=na.omit)
    leven_res<-leveneTest(col2,col1,data=df,na.action=na.omit)
    }
    return(bartlett_res)
}

Вышеуказанный скрипт является простой функцией, когда я запускаю его, получаю ошибку, как показано ниже:

> variance_homo(iris,'Species','Sepal.Length')
Error in bartlett.test.default("Sepal.Length", "Species") : 
  all observations are in the same group
Called from: bartlett.test.default("Sepal.Length", "Species")

В чем проблема ??

Answer 1

Вот пересмотренная версия, которая возвращает результаты обоих тестовых прогонов.

variance_homo <- function(df, col1, col2){
  if (is.numeric(df[[col2]]) && nlevels(df[[col1]]) >= 2){
    fmla <- paste(col2, col1, sep = '~')
    fmla <- as.formula(fmla)
    bartlett_res <- bartlett.test(fmla, df, na.action = na.omit)
    leven_res <- car::leveneTest(fmla, data = df, na.action = na.omit)
    list(bartlett = bartlett_res, levene = leven_res)
  }else stop('not enough levels.')
}

variance_homo(iris, 'Species', 'Sepal.Length')

Answer 2

Попробуйте метод двойных скобок:

variance_homo<-function(df,col1,col2){
    if (is.numeric(df[[col2]]) & nlevels(df[[col1]])>2){
    bartlett_res <- bartlett.test(df[[col2]]~df[[col1]],df,na.action=na.omit)
    leven_res <- leveneTest(df[[col2]],df[[col1]],data=df,na.action=na.omit)
}
return(bartlett_res)
}

Тогда:

variance_homo(iris,'Species','Sepal.Length')

    Bartlett test of homogeneity of variances

data:  df[[col2]] by df[[col1]]
Bartlett's K-squared = 16.006, df = 2, p-value = 0.0003345