r - 使用 purrr 映射将函数应用于 dplyr 管道中 DataFrame 中列的选择

Question

我有以下数据框

test <- data.frame(x = c(6, 9, 3, NA),
                   y = c(3, NA, 2, 3),
                   z = c(6, 3, NA, 5),
                   h = c(NA, 6, 7, 2))

这是我想遍历的列的列表

mylist <- list(test$y, test$z)

我想根据 ifelse 中的条件更改列“y”和“z”

这是我的尝试......似乎不起作用

test <- test %>%
          map_df(mylist, if(is.na(mylist), 0, 1))

(实际上我有一个更大的数据框，这只是测试数据)

我需要使用 mutate 吗？
我可以在管道中使用 select 吗？像这样？

test <- test %>%
          map_df(select(y, z), if(is.na(.), 0, 1))

这是预期的输出

test <- data.frame(x = c(6, 9, 3, NA),
                   y = c(1, 0, 1, 1),
                   z = c(1, 1, 0, 1),
                   h = c(NA, 6, 7, 2))

感谢帮助

Answer 1

我们可以使用mutate_at来指定列

library(dplyr)
test %>%  mutate_at(vars(y, z), ~as.integer(!is.na(.)))

#   x y z  h
#1  6 1 1 NA
#2  9 0 1  6
#3  3 1 0  7
#4 NA 1 1  2

或者如果 ifelse 是首选

test %>% mutate_at(vars(y, z), ~ifelse(is.na(.), 0, 1))

---

我们也可以在 base R 中做同样的事情

cols <- c("y", "z")
test[cols] <- as.integer(!is.na(test[cols]))