在多个条件下重新编码 dplyr

Question

我想将一个变量重新编码为 missing，如果它在 dplyr 中取三个值之一。考虑以下数据框 have:

id  married hrs_workperwk
1   1       40
2   1       55
3   1       70
4   0       -1
5   1       99
6   0       -2
7   0       10
8   0       40
9   1       45

-1、-2 和 99 是非法值。新数据框 want 应如下所示：

id  married hrs_workperwk
1   1       40
2   1       55
3   1       70
4   0       NA
5   1       NA
6   0       NA
7   0       10
8   0       40
9   1       45

我可以使用 base R 来快速解决这个问题，但是当我已经在使用 mutate() 时，dplyr 通常很方便。 las，这意味着我目前使用多个嵌套 if_else() 函数：

want <- mutate(have, 
hrs_workperwk = if_else(hrs_workperwk < 0, as.numeric(NA), 
                if_else(hrs_workperwk = 99, as.numeric(NA), hrs_workperwk)))

有没有办法只用一个 if_else() 函数来做到这一点？理想情况下是这样的：

want <- mutate(have, 
hrs_workperwk = if_else(hrs_workperwk = c(-2, -1, 99), as.numeric(NA), hrs_workperwk))

Answer 1

您可以使用 %in%:

want <- have %>% 
  mutate(hrs_workperwk = ifelse(hrs_workperwk %in% c(-1, -2, 99), NA, hrs_workperwk))

Answer 2

我们可以使用replace

df %>%
  mutate(hrs_workperwk = replace(hrs_workperwk, hrs_workperwk %in% c(-1, -2, 99), NA))
#  id married hrs_workperwk
#1  1       1            40
#2  2       1            55
#3  3       1            70
#4  4       0            NA
#5  5       1            NA
#6  6       0            NA
#7  7       0            10
#8  8       0            40
#9  9       1            45

或者另一种选择是case_when

df %>%
   mutate(hrs_workperwk = case_when(hrs_workperwk %in% c(-1, -2, 99)~ NA_integer_,
                      TRUE ~ hrs_workperwk))

Answer 3

在基数 R 中：

df1$hrs_workperwk[df1$hrs_workperwk %in% c(-1,-2,99)] <- NA

或

is.na(df1$hrs_workperwk) <- df1$hrs_workperwk %in% c(-1,-2,99)

两种情况的输出：

#   id married hrs_workperwk
# 1  1       1            40
# 2  2       1            55
# 3  3       1            70
# 4  4       0            NA
# 5  5       1            NA
# 6  6       0            NA
# 7  7       0            10
# 8  8       0            40
# 9  9       1            45

数据

df1 <- read.table(text="
id  married hrs_workperwk
1   1       40
2   1       55
3   1       70
4   0       -1
5   1       99
6   0       -2
7   0       10
8   0       40
9   1       45",h=T,strin=F)

在多个条件下重新编码 dplyr

Recode on multiple conditions dplyr

r

dplyr

recode