case_when 和 is.na 的意外行为

Question

我想将列中的所有 NA 值更改为 0，将所有其他值更改为 1。但是，我无法让 case_when 和 is.na 的组合起作用。

# Create dataframe
a <- c(rep(NA,9), 2, rep(NA, 10))
b <- c(rep(NA,9), "test", rep(NA, 10))
df <- data.frame(a,b, stringsAsFactors = F)

# Create new column (c), where all NA values in (a) are transformed to 0 and other values are transformed to 1
df <- df %>% 
  mutate(
    c = case_when(
      a == is.na(.$a) ~ 0,
      FALSE ~ 1
    )
  )

我希望列 (c) 指示所有 0 值和一个 1 值，但它全是 0。

当我在 is.na 中使用 if_else 语句时，它确实有效，例如：

df <- df %>% 
  mutate(
    c = if_else(is.na(a), 0, 1))
  )

这是怎么回事？

Answer 1

你应该这样做：

df %>% 
    mutate(
        c = case_when(
            is.na(a) ~ 0,
            TRUE ~ 1
        )
    )