避免与DPLYR::CASE_WHEN发生类型冲突

Question

我试图在dplyr::mutate中使用dplyr::case_when来创建一个新变量，在这个变量中，我将一些值设置为missing并同时重新编码其他值。

但是，如果尝试将值设置为NA，则会出现一个错误，说明我们无法创建变量new，因为NA是逻辑的：

mutate_impl(.data,dots）中出错:
计算错误：类型必须是double,不是逻辑。

在数据帧的非逻辑向量中，是否有方法使用此设置值na？

library(dplyr)    

# Create data
df <- data.frame(old = 1:3)

# Create new variable
df <- df %>% dplyr::mutate(new = dplyr::case_when(old == 1 ~ 5,
                                                  old == 2 ~ NA,
                                                  TRUE ~ old))

# Desired output
c(5, NA, 3)

Answer 1

如？case_when中所述：

所有RHS必须计算为相同类型的向量。

你实际上有两种可能：

df <- df %>% mutate(new = case_when(old == 1 ~ 5,
                                    old == 2 ~ NA_real_,
                                    TRUE ~ as.numeric(old)))

str(df)
# 'data.frame': 3 obs. of  2 variables:
# $ old: int  1 2 3
# $ new: num  5 NA 3

df <- df %>% mutate(new = case_when(old == 1 ~ 5L,
                                    old == 2 ~ NA_integer_,
                                    TRUE ~ old))

str(df)
# 'data.frame': 3 obs. of  2 variables:
# $ old: int  1 2 3
# $ new: int  5 NA 3