r - 使用具有不确定数量的列的 replace_na()

Question

我的数据框是这样的:

df <- tibble(x = c(1, 2, NA),
             y = c(1, NA, 3),
             z = c(NA, 2, 3))

我想使用 tidyr::replace_na() 将 NA 替换为 0。正如此函数的文档所阐明的那样，一旦您知道要对哪些列执行操作，就可以直接执行此操作。

df <- df %>% replace_na(list(x = 0, y = 0, z = 0))

但是如果列数不确定怎么办？ (我说“不确定”是因为我正在尝试创建一个使用 dplyr 工具即时执行此操作的函数。)如果我没记错的话，与我使用上述工具尝试实现的基本 R 等效的是:

df[, 1:ncol(df)][is.na(df[, 1:ncol(df)])] <- 0

但我总是难以理解这段代码。预先感谢您的帮助。

Answer 1

我们可以根据数据集的列数创建一个由 0 组成的列表，并使用列名设置名称

library(tidyverse)
df %>% 
   replace_na(set_names(as.list(rep(0, length(.))), names(.)))
# A tibble: 3 x 3
#      x     y     z
#   <dbl> <dbl> <dbl>
#1     1     1     0
#2     2     0     2
#3     0     3     3

或者另一种选择是 mutate_all(对于选定的列 -mutate_at 或基础 don 条件 mutate_if)和 applyreplace_all

df %>%
    mutate_all(replace_na, replace = 0)

---

有了base R，就更直接了

df[is.na(df)] <- 0