Mutate dplyr выбрать первое значение переменной

Question

У меня есть dataframe, который выглядит, как этот

test<-data.frame(x=c(100,100,101,101,102,102),y=c(1,2,1,2,1,2),z=c(NA,NA,0.1,NA,NA,0.5)) 

Я хочу, чтобы сгруппировать х. Затем создайте новую переменную ynew, где я бы посмотрел на z и выберем первое значение, которое не является NA, а затем установите ynew = y. Если оба значения в z равны NA, я хотел бы, чтобы yNew был NA. ynew должно быть NA, NA, 1,1,2,2. Я пытаюсь сделать это с помощью dplyr.

Я застрял со следующим

group_by(test,x) %>% mutate(ynew=ifelse(all(is.na(z)),NA_integer_,y[corresponding index])) 

Answer 1

Следующая dplyr подход работает, если test является data.table

library(data.table) 
test <- data.table(test) 

test %>% group_by(x) %>% mutate(ynew = first(y[!is.na(z)])) 

# Source: local data table [6 x 4] 

#  x  y  z ynew 
# (dbl) (dbl) (dbl) (dbl) 
#1 100  1 NA NA 
#2 100  2 NA NA 
#3 101  1 0.1  1 
#4 101  2 NA  1 
#5 102  1 NA  2 
#6 102  2 0.5  2 

Или вы можете просто использовать data.table путь

test[, ynew := y[!is.na(z)], x]