在 R 中用 0 代替 NA

Gustavo du Mortier 2023年1月30日
  1. 在较大的 R 数据框中用零代替 NA
  2. 在 R 数据框的子集中用零替换 NA
在 R 中用 0 代替 NA

在 R 中,有一个简单的方法可以将数据框中的 NA 替换为 0。假设你有一个名为 my_data 的数据框。要将该数据框中所有的 NA 值用零代替,可以执行这条语句。

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

例如,如果 my_data 有以下内容。

   C1  C2    C3  C4  C5
1   4   3  <NA>   3   7
2   9   8   ABC   5  10
3   1   1   XYZ   3   6
4  NA   4  <NA>   7  10
5   1   2   ZC1  NA   2

当你执行 my_data[is.na(my_data)] <- 0 时,数据框的内容就会变成这样。

   C1  C2    C3  C4  C5
1   4   3     0   3   7
2   9   8   ABC   5  10
3   1   1   XYZ   3   6
4   0   4     0   7  10
5   1   2   ZC1   0   2

在较大的 R 数据框中用零代替 NA

以前的解决方案使用 Base R 子集重分配,当你有相对较小的数据框时,它工作得很好。但对于更大的数据集,你可能需要一个更快的替代方案,比如最近版本的 dplyr 包中实现的新的混合评估方法。

dplyr 包采用的新方法可以识别整个表达式,并使用 C++ 代码来评估它们。这样,在处理大数据框时,你可以实现高达 30%的变换速度。

要使用 dplyr 包将 NA 值替换为零,可以使用 mutate 函数与 _all 范围内的动词和 replace 函数的 purrr 格式,如下例所示。

my_data <- mutate_all(my_data, ~replace(., is.na(.), 0))

使用 purrr 符号允许我们将 replace 函数应用于每个数据框元素。

在 R 数据框的子集中用零替换 NA

mutate 函数中,你可以使用 _at 范围动词来代替 _all 范围动词,将替换操作限制在特定的列上。要做到这一点,你可以包含一个包含列名的向量,这是你希望替换发生的地方。使用前面的数据框,如果你只需要替换列 C1C4 中的 NA 值,你可以使用以下命令。

my_data <- mutate_at(my_data, c(`C1`, `C4`), ~replace(., is.na(.), 0))

这样一来,只有列 C1C4 中的 NA 值被替换为 0,从而得到如下数据框。

   C1  C2    C3  C4  C5
1   4   3  <NA>   3   7
2   9   8   ABC   5  10
3   1   1   XYZ   3   6
4   0   4  <NA>   7  10
5   1   2   ZC1   0   2

在前面的例子中,你可能希望只在数字列中用 0 来替换 NA,以避免在字母数字列(如 C3)中包含 0 值。如果是这种情况,你可以使用 mutate_if 函数和 is.numeric 条件来告诉 R 只在数字列中用零值替换 NA,而不是指定要应用替换的列。在下面的例子中,你可以找到完整的代码来尝试,从安装 dplyr 包、填充数据框到执行替换并显示结果。

install.packages("dplyr")
library(dplyr)
C1 <- c(4, 9, 1, NA, 1)
C2 <- c(3, 8, 1, 4, 2)
C3 <- c(NA, 'ABC', 'XYZ', NA, 'ZC1')
C4 <- c(3, 5, 3, 7, NA)
C5 <- c(7, 10, NA, 10, 2)
my_data <- data.frame(C1, C2, C3, C4, C5)
my_data <- mutate_if(my_data, is.numeric, ~replace(., is.na(.), 0))
my_data

输出:

   C1  C2    C3  C4  C5
1   4   3  <NA>   3   7
2   9   8   ABC   5  10
3   1   1   XYZ   3   0
4   0   4  <NA>   7  10
5   1   2   ZC1   0   2

你可以在 R 文档中找到更多关于 mutate() 函数及其变体的信息。

相关文章 - R Data Frame