r - 根据 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