一个简单的R语言数据分析案例

news/2024/10/20 4:02:24/

在R语言中，数据分析可以涵盖广泛的领域，包括描述性统计、探索性数据分析、假设检验、数据可视化、机器学习等。以下是一个简单的R语言数据分析案例，该案例将涵盖数据导入、数据清洗、描述性统计、数据可视化以及一个简单的预测模型。

案例：预测房价

假设我们有一个关于房价的数据集，其中包含房屋的面积、卧室数量、浴室数量、楼层、房屋类型等信息以及房价。我们的目标是使用这些特征来预测房价。

步骤 1: 数据导入

首先，我们需要导入数据集。在这个例子中，我们将使用CSV文件来存储数据。

# 导入数据
data <- read.csv("house_prices.csv")# 查看数据的前几行
head(data)

步骤 2: 数据清洗

在导入数据后，我们需要进行数据清洗，以确保数据的准确性和一致性。这可能包括处理缺失值、异常值、数据转换等。

# 处理缺失值（这里假设我们用均值填充缺失的数值型特征）
data$area[is.na(data$area)] <- mean(data$area, na.rm = TRUE)# 转换分类变量为数值型（例如，使用独热编码或标签编码）
# 这里我们假设楼层是一个分类变量，我们可以将其转换为数值型
data$floor <- as.numeric(as.factor(data$floor))# 删除不必要的列或添加新的列（如果有需要的话）
# ...

步骤 3: 描述性统计

接下来，我们可以对数据进行描述性统计分析，以了解数据的整体特征。

# 计算数值型特征的描述性统计信息
summary(data[sapply(data, is.numeric)])# 对于分类变量，可以使用表格来查看分布
table(data$house_type)

步骤 4: 数据可视化

数据可视化可以帮助我们更直观地理解数据。

# 绘制房价与面积的关系图
plot(data$area, data$price, xlab = "Area (sq ft)", ylab = "Price ($)", main = "Price vs Area")# 使用箱线图查看不同房屋类型的价格分布
boxplot(price ~ house_type, data = data, main = "Price Distribution by House Type", xlab = "House Type", ylab = "Price ($)")

步骤 5: 预测模型

最后，我们可以使用机器学习算法来构建预测模型。在这个例子中，我们将使用简单的线性回归模型来预测房价。

# 分离特征和目标变量
X <- data[, !(names(data) %in% "price")]  # 特征
y <- data$price  # 目标变量# 划分训练集和测试集（这里使用随机划分，但实际应用中建议使用交叉验证或时间分割）
set.seed(123)  # 设置随机种子以便结果可复现
train_idx <- sample(1:nrow(data), 0.7 * nrow(data))
train_data <- data[train_idx, ]
test_data <- data[-train_idx, ]# 在训练集上训练线性回归模型
library(stats)
model <- lm(price ~ ., data = train_data)# 查看模型摘要以获取系数、R平方等统计信息
summary(model)# 在测试集上进行预测并评估模型性能
predictions <- predict(model, newdata = test_data)
mse <- mean((predictions - test_data$price)^2)  # 计算均方误差
print(paste("Mean Squared Error:", mse))