根据 dplyr 中的条件更改多个列值

Question

我正在寻找一种简单的方法来执行以下操作，但使用 dplyr，基本上只是在满足条件时用 NA 替换 3 列中的值。

dta[dta$na.ind == 1, c('x1', 'x2', 'x3')] <- NA

我能想到的使用 dplyr 的唯一方法如下，但我觉得应该有一个更简单的方法

dta <- dta %>% 
    mutate(x1 = ifelse(na.ind == 1, NA, x1),
           x2 = ifelse(na.ind == 1, NA, x2),
           x3 = ifelse(na.ind == 1, NA, x3))

谢谢！

Answer 1

您可以使用mutate_at列并将其传递x1,x2,x3给.vars参数：

dta <- data.frame(na.ind = 1:3, x1 = 2:4, x2 = 2:4, x3 = 2:4, x4 = 2:4)
dta
#  na.ind x1 x2 x3 x4
#1      1  2  2  2  2
#2      2  3  3  3  3
#3      3  4  4  4  4

dta %>% mutate_at(.vars = c("x1", "x2", "x3"), funs(ifelse(na.ind == 1, NA, .)))
#  na.ind x1 x2 x3 x4
#1      1 NA NA NA  2
#2      2  3  3  3  3
#3      3  4  4  4  4

Answer 2

mutate_at由于诸如等作用域动词mutate_all被 替换across，您现在可以对 Psidom 的答案进行以下调整：

> dta <- data.frame(na.ind = 1:3, x1 = 2:4, x2 = 2:4, x3 = 2:4, x4 = 2:4)
> dta
  na.ind x1 x2 x3 x4
1      1  2  2  2  2
2      2  3  3  3  3
3      3  4  4  4  4

> dta %>% mutate(across(c(x1, x2, x3), ~ ifelse(na.ind == 1, NA, .)))
  na.ind x1 x2 x3 x4
1      1 NA NA NA  2
2      2  3  3  3  3
3      3  4  4  4  4

有关 的更多信息，请参阅https://dplyr.tidyverse.org/reference/across.htmlacross。