使用R语言编写自定义函数以均值填充每个数据列的缺失值

使用R语言编写自定义函数以均值填充每个数据列的缺失值

在数据分析和处理中，经常会遇到数据集中存在缺失值的情况。处理缺失值的一种常见方法是使用均值填充。在R语言中，我们可以编写自定义函数来实现这个功能。本文将介绍如何使用R语言编写一个自定义函数，以均值填充每个数据列的缺失值。

首先，让我们定义一个示例数据集，其中包含一些缺失值：

# 创建示例数据集
data <- data.frame(
  x1 = c(1, 2, NA, 4, 5),
  x2 = c(NA, 2, 3, NA, 5),
  x3 = c(1, 2, 3, 4, NA)
)

接下来，我们可以编写一个名为mean_impute的自定义函数，该函数将接受一个数据框作为输入，并返回填充了均值的数据框。以下是该函数的代码：

# 自定义函数：均值填充缺失值
mean_impute <- function(data) {
  for (col in colnames(data)) {
    # 计算每列的均值
    col_mean <- mean(data[[col]], na.rm = TRUE)
    
    # 使用均值填充缺失值
    data[[col]][is.na(data[[col]])] <- col_mean
  }