использовать пользовательскую функцию после сгруппированных данных в R

У меня есть функция, проверяющая нулевые числа в каждом столбце в большом кадре данных. Теперь я хочу проверить нулевые числа в каждом столбце после группировки по категориям. Вот пример:

   zero_rate <- function(df) {
     z_rate_list <- sapply(df, function(x) {
      data.frame(
      n_zero=length(which(x==0)), 
      n=length(x), 
      z_rate=length(which(x==0))/length(x))
 })

      d <- data.frame(z_rate_list)
      d <- sapply(d, unlist)
      d <- as.data.frame(d)

      return(d)}

   df = data.frame(var1=c(1,0,NA,4,NA,6,7,0,0,10),var2=c(11,NA,NA,0,NA,16,0,NA,19,NA))
   df1= data.frame(cat = c(1,1,1,1,1,2,2,2,2,2),df)


   zero_rate_df =  df1 %>% group_by(cat) %>% do( zero_rate(.))

Здесь zero_rate(df) работает так, как я и ожидал. Но когда я группирую данные по коту и вычисляю в каждой категории нулевую скорость для каждого столбца, результат не такой, как я ожидал. Я ожидаю что-то вроде этого:

   cat         va1  var2
    1   n_zero  1   1
            n   5   5
        z_rate  0.2 0.2
    2   n_zero  2   1
            n   5   5
       z_rate   0.4 0.2

Любое предложение? Спасибо.


r group-by user-defined
person newleaf    schedule 05.04.2017    source источник

Ответы (1)


arrow_upward
3
arrow_downward

Я придумал следующий код. .[-1] использовался для удаления столбца группировки:

zero_rate <- function(df){
    res <- lapply(df, function(x){
        y <- c(sum(x == 0, na.rm = T), length(x))
        c(y, y[1]/y[2])
    })
    res <- do.call(cbind.data.frame, res)
    res$vars <- c('n_zero', 'n', 'z_rate')
    res
}

df1 %>% group_by(cat) %>% do( zero_rate(.[-1]))

#     cat  var1  var2   vars
#   <dbl> <dbl> <dbl>  <chr>
# 1     1   1.0   1.0 n_zero
# 2     1   5.0   5.0      n
# 3     1   0.2   0.2 z_rate
# 4     2   2.0   1.0 n_zero
# 5     2   5.0   5.0      n
# 6     2   0.4   0.2 z_rate
person mt1022    schedule 05.04.2017
comment
Спасибо за улучшение моей части. sapply +unlist является избыточным. - person newleaf; 05.04.2017