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

Question

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

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

有关 across的更多信息，请参见

https://dplyr.tidyverse.org/reference/across.html

。

Answer 3

如果您想要在 NA 的任何列中按 data.frame-wise 替换特定值 (-99999)：

dta %>% mutate_all(funs(ifelse(.==-99999, NA, .)))

更新：

funs

已弃用。上面的代码现在是：

dta %>% mutate(across(c(x1, x2, x3), ~ ifelse(.==-99999, NA, .)))