Почему фильтр dplyr оставляет значения NA из переменной фактора?

Question

Когда я использую filter из пакета dplyr, чтобы понизить уровень факторной переменной, filter также снижает значения NA. Вот пример:

library(dplyr) 
set.seed(919) 
(dat <- data.frame(var1 = factor(sample(c(1:3, NA), size = 10, replace = T)))) 
# var1 
# 1 <NA> 
# 2  3 
# 3  3 
# 4  1 
# 5  1 
# 6 <NA> 
# 7  2 
# 8  2 
# 9 <NA> 
# 10 1 

filter(dat, var1 != 1) 
# var1 
# 1 3 
# 2 3 
# 3 2 
# 4 2 

Это не кажется идеальным - я только хотел бросить строки, где var1 == 1.

Похоже, что это происходит потому, что любой comparison with NA returns NA, который затем падает filter. Так, например, filter(dat, !(var1 %in% 1)) дает правильные результаты. Но есть ли способ сказать filter, чтобы не уронить значения NA?

Answer 1

Вы могли бы использовать это:

filter(dat, var1 != 1 | is.na(var1)) 
    var1 
1 <NA> 
2 3 
3 3 
4 <NA> 
5 2 
6 2 
7 <NA> 

И это не будет.

Кроме того, только для завершения, сбросив NAs является намеченная поведение filter как вы можете видеть из следующего:

test_that("filter discards NA", { 
    temp <- data.frame(
    i = 1:5, 
    x = c(NA, 1L, 1L, 0L, 0L) 
) 
    res <- filter(temp, x == 1) 
    expect_equal(nrow(res), 2L) 
}) 

Этот тест выше был взят из испытаний для filter от github.