Как не заставить dplyr mutate вычислять для отсутствующей группы

Question

У меня есть данные, которые выглядят так:

df=data.frame(a=1:6,b=rep(c("one","two"),each=3))
df[2,2]<-NA

Я хочу вычислить среднее значение каждой группы для каждой строки, например:

df %>% group_by(b) %>% mutate(mean=mean(a))

Проблема в том, что R просматриваетАН как группа.Желаемый результат будет

среднее = c (2, NA, 2,5,5,5).

Попытка :

df %>% group_by(b) %>% mutate(mean=if_else(b==NA,NA,mean(a)))

но это выдает ошибку

Answer 1

Попробуйте

df %>% group_by(b) %>% mutate(mean=mean(a)) %>% mutate(mean = if_else(is.na(b), NA_real_, mean))

Answer 2

Если вы хотите избежать сообщений об ошибках:

library(hablar)

df %>% 
  convert(chr(b)) %>% 
  group_by(b) %>% 
  mutate(mean = if_else_(!is.na(b), mean(a), NA))